message: Set size of test //third_party/tensorflow/python/data/kernel_tests:dataset_from_generator_op_test to medium
  It sometimes takes longer than a minute, and thus gets flaky timeouts.
diff: @@ -59,7 +59,7 @@ tf_py_test(
  tf_py_test(
  name = "dataset_from_generator_op_test",
  - size = "small",
  + size = "medium",
  srcs = ["dataset_from_generator_op_test.py"],
  additional_deps = [
  "//third_party/py/numpy",
  