experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[tfdbg] Mark the reasons for a number of tests decorated with run_v1_only

Browse Source 
PiperOrigin-RevId: 336726372
Change-Id: Iff0197fca0a8e53fc164f9d9b1fa969f1475104b
This commit is contained in:
Shanqing Cai 2020-10-12 13:03:52 -07:00 committed by TensorFlower Gardener
parent cf67a717ef
commit 117fde63a8
6 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/python/debug/cli/cli_shared_test.py
View File

@ -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):

6
tensorflow/python/debug/lib/common_test.py
View File

@ -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")

2
tensorflow/python/debug/lib/debug_gradients_test.py
View File

@ -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):

4
tensorflow/python/debug/lib/source_utils_test.py
View File

@ -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):

2
tensorflow/python/debug/wrappers/disk_usage_test.py
View File

@ -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

2
tensorflow/python/debug/wrappers/framework_test.py
View File

@ -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):