From cb23a88fe1c9372559db18202c562243bb855b9c Mon Sep 17 00:00:00 2001 From: Andrew Audibert Date: Wed, 14 Oct 2020 13:10:49 -0700 Subject: [PATCH] [tf.data service] Prevent flakiness in unit test. PiperOrigin-RevId: 337156045 Change-Id: I4598a8ae96a14be471c66826cfaea1966b3b2e9e --- tensorflow/python/data/experimental/kernel_tests/BUILD | 1 - .../data/experimental/kernel_tests/data_service_ops_test.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/tensorflow/python/data/experimental/kernel_tests/BUILD b/tensorflow/python/data/experimental/kernel_tests/BUILD index 0ea339d6301..248421b083f 100644 --- a/tensorflow/python/data/experimental/kernel_tests/BUILD +++ b/tensorflow/python/data/experimental/kernel_tests/BUILD @@ -157,7 +157,6 @@ tf_py_test( srcs = ["data_service_ops_test.py"], shard_count = 10, srcs_version = "PY3", - tags = ["notap"], # TODO(b/170783829) deps = [ ":data_service_test_base", "//tensorflow:tensorflow_py", diff --git a/tensorflow/python/data/experimental/kernel_tests/data_service_ops_test.py b/tensorflow/python/data/experimental/kernel_tests/data_service_ops_test.py index 65674818daf..ddd301d1540 100644 --- a/tensorflow/python/data/experimental/kernel_tests/data_service_ops_test.py +++ b/tensorflow/python/data/experimental/kernel_tests/data_service_ops_test.py @@ -214,7 +214,7 @@ class DataServiceOpsTest(data_service_test_base.TestBase, @combinations.generate(test_base.eager_only_combinations()) def testSharedJobName(self): cluster = self.create_cluster(num_workers=1) - num_elements = 100 + num_elements = 1000 def make_ds(): return dataset_ops.Dataset.range(num_elements).shuffle(num_elements)