Change test size to medium on map_and_batch test that was timing out.

PiperOrigin-RevId: 389243301 Change-Id: Ia3dc0265a54fab7bdf5b9b1aee9dd52d8667aadd
Marbck · Aug 6, 2021 · 12b3780 · 12b3780
1 parent b45c040
commit 12b3780
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow/python/data/experimental/kernel_tests/BUILD b/tensorflow/python/data/experimental/kernel_tests/BUILD
@@ -335,7 +335,7 @@ tf_py_test(
 
 tf_py_test(
     name = "map_and_batch_test",
-    size = "small",
+    size = "medium",
     srcs = ["map_and_batch_test.py"],
     shard_count = 4,
     deps = [