diff --git a/tensorflow/python/eager/BUILD b/tensorflow/python/eager/BUILD index b50d06c0d53..c08cb8cc1c3 100644 --- a/tensorflow/python/eager/BUILD +++ b/tensorflow/python/eager/BUILD @@ -960,7 +960,6 @@ cuda_py_test( shard_count = 8, tags = [ "no_oss", # This test launches local server - "notap", # TODO(b/155407846) ], deps = [ "//tensorflow/python:array_ops", diff --git a/tensorflow/python/eager/remote_cluster_test.py b/tensorflow/python/eager/remote_cluster_test.py index a0ce235c282..025899e271e 100644 --- a/tensorflow/python/eager/remote_cluster_test.py +++ b/tensorflow/python/eager/remote_cluster_test.py @@ -300,7 +300,6 @@ class DynamicClusterTest(test.TestCase, parameterized.TestCase): y = worker_fn(x1) np.testing.assert_array_equal([[2, 2], [2, 2]], y.numpy()) - @test_util.run_in_async_and_sync_mode def testPendingNodesServerReplaced(self): """Update cluster when nodes are still pending on remote workers.""" with ops.device(self.device_local): @@ -371,10 +370,8 @@ class DynamicClusterTest(test.TestCase, parameterized.TestCase): for result in t1_results + t2_results: np.testing.assert_array_equal([[2, 2], [2, 2]], result) - @test_util.run_in_async_and_sync_mode def testMultiThreadPendingNodesLockFree(self): """Update cluster when other remote function calls are being launched.""" - self.skipTest("b/154053481") with ops.device(self.device_t1): x1 = array_ops.ones([2, 2])