diff options
Diffstat (limited to 'tensorflow/python/debug/wrappers/dumping_wrapper_test.py')
-rw-r--r-- | tensorflow/python/debug/wrappers/dumping_wrapper_test.py | 18 |
1 files changed, 1 insertions, 17 deletions
diff --git a/tensorflow/python/debug/wrappers/dumping_wrapper_test.py b/tensorflow/python/debug/wrappers/dumping_wrapper_test.py index acea9433e2..d987ba84b5 100644 --- a/tensorflow/python/debug/wrappers/dumping_wrapper_test.py +++ b/tensorflow/python/debug/wrappers/dumping_wrapper_test.py @@ -111,20 +111,6 @@ class DumpingDebugWrapperSessionTest(test_util.TensorFlowTestCase): self.assertEqual(repr(self.inc_v), dump.run_fetches_info) self.assertEqual(repr(None), dump.run_feed_keys_info) - def testDumpingOnASingleRunWorksWithRelativePathForDebugDumpDir(self): - sess = dumping_wrapper.DumpingDebugWrapperSession( - self.sess, session_root=self.session_root, log_usage=False) - sess.run(self.inc_v) - dump_dirs = glob.glob(os.path.join(self.session_root, "run_*")) - cwd = os.getcwd() - try: - os.chdir(self.session_root) - dump = debug_data.DebugDumpDir( - os.path.relpath(dump_dirs[0], self.session_root)) - self.assertAllClose([10.0], dump.get_tensors("v", 0, "DebugIdentity")) - finally: - os.chdir(cwd) - def testDumpingOnASingleRunWithFeedDictWorks(self): sess = dumping_wrapper.DumpingDebugWrapperSession( self.sess, session_root=self.session_root, log_usage=False) @@ -364,14 +350,12 @@ class DumpingDebugWrapperSessionTest(test_util.TensorFlowTestCase): thread_name_filter=r"MainThread$") self.assertAllClose(1.0, sess.run(self.delta)) - child_thread_result = [] def child_thread_job(): - child_thread_result.append(sess.run(self.eta)) + sess.run(sess.run(self.eta)) thread = threading.Thread(name="ChildThread", target=child_thread_job) thread.start() thread.join() - self.assertAllClose([-1.4], child_thread_result) dump_dirs = glob.glob(os.path.join(self.session_root, "run_*")) self.assertEqual(1, len(dump_dirs)) |