experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[tf.data service] Prevent flakiness in unit test.

Browse Source 
PiperOrigin-RevId: 337156045
Change-Id: I4598a8ae96a14be471c66826cfaea1966b3b2e9e
This commit is contained in:
Andrew Audibert 2020-10-14 13:10:49 -07:00 committed by TensorFlower Gardener
parent 0e0522275b
commit cb23a88fe1
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/python/data/experimental/kernel_tests/BUILD
View File

@ -157,7 +157,6 @@ tf_py_test(
srcs = ["data_service_ops_test.py"], srcs = ["data_service_ops_test.py"],
shard_count = 10, shard_count = 10,
srcs_version = "PY3", srcs_version = "PY3",
tags = ["notap"], # TODO(b/170783829)
deps = [ deps = [
":data_service_test_base", ":data_service_test_base",
"//tensorflow:tensorflow_py", "//tensorflow:tensorflow_py",

2
tensorflow/python/data/experimental/kernel_tests/data_service_ops_test.py
View File

@ -214,7 +214,7 @@ class DataServiceOpsTest(data_service_test_base.TestBase,
@combinations.generate(test_base.eager_only_combinations()) @combinations.generate(test_base.eager_only_combinations())
def testSharedJobName(self): def testSharedJobName(self):
cluster = self.create_cluster(num_workers=1) cluster = self.create_cluster(num_workers=1)
num_elements = 100 num_elements = 1000
def make_ds(): def make_ds():
return dataset_ops.Dataset.range(num_elements).shuffle(num_elements) return dataset_ops.Dataset.range(num_elements).shuffle(num_elements)