#tf-data-service Make sure dynamic sharding works without job_names.

PiperOrigin-RevId: 564516421

#tf-data-service Make sure dynamic sharding works without job_names.
PiperOrigin-RevId: 564516421
4551e3ef · Yang Chen · TensorFlower Gardener · da711425 · 4551e3ef
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

tensorflow/python/data/experimental/kernel_tests/service/dynamic_sharding_test.py ...xperimental/kernel_tests/service/dynamic_sharding_test.py +2 -1

未找到文件。
--- a/tensorflow/python/data/experimental/kernel_tests/service/dynamic_sharding_test.py
+++ b/tensorflow/python/data/experimental/kernel_tests/service/dynamic_sharding_test.py
@@ -57,7 +57,8 @@ class DynamicShardingTest(data_service_test_base.TestBase,
    ds = self.make_distributed_dataset(
        ds,
        cluster,
-        processing_mode=data_service_ops.ShardingPolicy.DYNAMIC)
+        processing_mode=data_service_ops.ShardingPolicy.DYNAMIC,
+        job_name=None)
    self.assertDatasetProduces(
        ds, list(range(num_elements)), assert_items_equal=True)