From 501a65c3469bfeafb508c97db0654ad694460167 Mon Sep 17 00:00:00 2001 From: Mihai Maruseac Date: Mon, 9 May 2022 07:49:44 -0700 Subject: [PATCH] Disable timing out tests --- tensorflow/python/data/kernel_tests/BUILD | 6 ++++++ tensorflow/python/kernel_tests/linalg/sparse/BUILD | 1 + 2 files changed, 7 insertions(+) diff --git a/tensorflow/python/data/kernel_tests/BUILD b/tensorflow/python/data/kernel_tests/BUILD index 2fb8e112a280e5..f6b64b0b9fe453 100644 --- 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 index 91c8d0f8cdc356..036f739b2c7154 100644 --- 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 = [