Update the test cases for testNumThreadsDeprecated

feihugis · Dec 3, 2019 · c45ea87 · c45ea87
1 parent 031e618
commit c45ea87
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tensorflow/python/data/experimental/kernel_tests/override_threadpool_test.py b/tensorflow/python/data/experimental/kernel_tests/override_threadpool_test.py
@@ -75,9 +75,7 @@ def get_thread_id(_):
           combinations.combine(
               num_threads=[1, 2, 4, 8, 16], max_intra_op_parallelism=[None]) +
           combinations.combine(
-              num_threads=[4], max_intra_op_parallelism=[0, 1, 4]) +
-          combinations.combine(
-              num_threads=[5], max_intra_op_parallelism=[-1])))
+              num_threads=[4], max_intra_op_parallelism=[-1, 0, 4])))
   def testNumThreadsDeprecated(self, num_threads, max_intra_op_parallelism):
 
     def override_threadpool_fn(dataset):