Disable timing out tests

mihaimaruseac · mihaimaruseac · commit 501a65c3469b · 2022-05-09T07:49:44.000-07:00
diff --git a/tensorflow/python/data/kernel_tests/BUILD b/tensorflow/python/data/kernel_tests/BUILD
@@ -68,6 +68,9 @@ tf_py_test(
     name = "cache_test",
     size = "small",
     srcs = ["cache_test.py"],
+    tags = [
+        "no_oss",  # Times out during patch release
+    ],
     deps = [
         ":checkpoint_test_base",
         ":test_base",
@@ -536,6 +539,9 @@ tf_py_test(
     size = "small",
     srcs = ["map_test.py"],
     shard_count = 19,
+    tags = [
+        "no_oss",  # Times out during patch release
+    ],
     deps = [
         ":checkpoint_test_base",
         ":test_base",
diff --git a/tensorflow/python/kernel_tests/linalg/sparse/BUILD b/tensorflow/python/kernel_tests/linalg/sparse/BUILD
@@ -44,6 +44,7 @@ cuda_py_test(
     main = "csr_sparse_matrix_ops_test.py",
     shard_count = 10,
     tags = [
+        "no_oss",  # Times out during patch release
         "notsan",  # b/149115441
     ],
     deps = [
