diff --git a/tensorflow/python/debug/cli/cli_shared_test.py b/tensorflow/python/debug/cli/cli_shared_test.py index 3182d29d125..2d461c9e630 100644 --- a/tensorflow/python/debug/cli/cli_shared_test.py +++ b/tensorflow/python/debug/cli/cli_shared_test.py @@ -105,7 +105,7 @@ class TimeToReadableStrTest(test_util.TensorFlowTestCase): cli_shared.time_to_readable_str(100, force_time_unit="ks") -@test_util.run_deprecated_v1 +@test_util.run_v1_only("tfdbg CLI is for tf.Session only") class GetRunStartIntroAndDescriptionTest(test_util.TensorFlowTestCase): def setUp(self): @@ -319,7 +319,7 @@ class GetRunStartIntroAndDescriptionTest(test_util.TensorFlowTestCase): self.assertEqual("run #1: 1 fetch (a:0); 1 feed (foo)", short_description) -@test_util.run_deprecated_v1 +@test_util.run_v1_only("tfdbg CLI is for tf.Session only") class GetErrorIntroTest(test_util.TensorFlowTestCase): def setUp(self): diff --git a/tensorflow/python/debug/lib/common_test.py b/tensorflow/python/debug/lib/common_test.py index f6413f6b7b3..8469700accf 100644 --- a/tensorflow/python/debug/lib/common_test.py +++ b/tensorflow/python/debug/lib/common_test.py @@ -27,7 +27,7 @@ from tensorflow.python.platform import googletest class CommonTest(test_util.TensorFlowTestCase): - @test_util.run_deprecated_v1 + @test_util.run_v1_only("Relies on tensor name, which is unavailable in TF2") def testOnFeedOneFetch(self): a = constant_op.constant(10.0, name="a") b = constant_op.constant(20.0, name="b") @@ -36,7 +36,7 @@ class CommonTest(test_util.TensorFlowTestCase): self.assertItemsEqual(["a:0"], loaded[0]) self.assertItemsEqual(["b:0"], loaded[1]) - @test_util.run_deprecated_v1 + @test_util.run_v1_only("Relies on tensor name, which is unavailable in TF2") def testGetRunKeyFlat(self): a = constant_op.constant(10.0, name="a") b = constant_op.constant(20.0, name="b") @@ -45,7 +45,7 @@ class CommonTest(test_util.TensorFlowTestCase): self.assertItemsEqual(["a:0"], loaded[0]) self.assertItemsEqual(["a:0", "b:0"], loaded[1]) - @test_util.run_deprecated_v1 + @test_util.run_v1_only("Relies on tensor name, which is unavailable in TF2") def testGetRunKeyNestedFetches(self): a = constant_op.constant(10.0, name="a") b = constant_op.constant(20.0, name="b") diff --git a/tensorflow/python/debug/lib/debug_gradients_test.py b/tensorflow/python/debug/lib/debug_gradients_test.py index 95da6cb9ff8..e7a4d2d5450 100644 --- a/tensorflow/python/debug/lib/debug_gradients_test.py +++ b/tensorflow/python/debug/lib/debug_gradients_test.py @@ -36,7 +36,7 @@ from tensorflow.python.platform import googletest from tensorflow.python.training import gradient_descent -@test_util.run_deprecated_v1 +@test_util.run_v1_only("Sessions are not available in TF 2.x") class IdentifyGradientTest(test_util.TensorFlowTestCase): def setUp(self): diff --git a/tensorflow/python/debug/lib/source_utils_test.py b/tensorflow/python/debug/lib/source_utils_test.py index d0a4ecdbac4..366b25e89ac 100644 --- a/tensorflow/python/debug/lib/source_utils_test.py +++ b/tensorflow/python/debug/lib/source_utils_test.py @@ -106,7 +106,7 @@ class GuessIsTensorFlowLibraryTest(test_util.TensorFlowTestCase): self.assertTrue( source_utils.guess_is_tensorflow_py_library(source_utils.__file__)) - @test_util.run_deprecated_v1 + @test_util.run_v1_only("Tensor.op is not available in TF 2.x") def testFileInPythonKernelsPathReturnsTrue(self): x = constant_op.constant(42.0, name="x") self.assertTrue( @@ -320,7 +320,7 @@ class SourceHelperTest(test_util.TensorFlowTestCase): source_utils.load_source(source_path) -@test_util.run_v1_only("b/120545219") +@test_util.run_v1_only("Sessions are not available in TF 2.x") class ListSourceAgainstDumpTest(test_util.TensorFlowTestCase): def createAndRunGraphWithWhileLoop(self): diff --git a/tensorflow/python/debug/wrappers/disk_usage_test.py b/tensorflow/python/debug/wrappers/disk_usage_test.py index 71c56b33106..e9cb388c89f 100644 --- a/tensorflow/python/debug/wrappers/disk_usage_test.py +++ b/tensorflow/python/debug/wrappers/disk_usage_test.py @@ -32,7 +32,7 @@ from tensorflow.python.platform import googletest from tensorflow.python.training import monitored_session -@test_util.run_deprecated_v1 +@test_util.run_v1_only("Sessions are not available in TF 2.x") class DumpingDebugWrapperDiskUsageLimitTest(test_util.TensorFlowTestCase): @classmethod diff --git a/tensorflow/python/debug/wrappers/framework_test.py b/tensorflow/python/debug/wrappers/framework_test.py index 266404a1038..7ca0cf192fa 100644 --- a/tensorflow/python/debug/wrappers/framework_test.py +++ b/tensorflow/python/debug/wrappers/framework_test.py @@ -141,7 +141,7 @@ class TestDebugWrapperSessionBadAction(framework.BaseDebugWrapperSession): return framework.OnRunEndResponse() -@test_util.run_deprecated_v1 +@test_util.run_v1_only("Sessions are not available in TF 2.x") class DebugWrapperSessionTest(test_util.TensorFlowTestCase): def _no_rewrite_session_config(self):