diff options
Diffstat (limited to 'tensorflow/python/training/session_manager_test.py')
-rw-r--r-- | tensorflow/python/training/session_manager_test.py | 28 |
1 files changed, 14 insertions, 14 deletions
diff --git a/tensorflow/python/training/session_manager_test.py b/tensorflow/python/training/session_manager_test.py index d7e6dac95b..f1d18f7704 100644 --- a/tensorflow/python/training/session_manager_test.py +++ b/tensorflow/python/training/session_manager_test.py @@ -98,7 +98,7 @@ class SessionManagerTest(test.TestCase): os.rename(checkpoint_dir, checkpoint_dir2) gfile.MakeDirs(checkpoint_dir) v = variables.Variable([6.0, 7.0, 8.0], name="v") - with self.test_session(): + with self.cached_session(): self.assertEqual(False, variables.is_variable_initialized(v).eval()) session_manager.SessionManager( ready_op=variables.report_uninitialized_variables()) @@ -236,7 +236,7 @@ class SessionManagerTest(test.TestCase): trainable=False, collections=[ops.GraphKeys.LOCAL_VARIABLES], name="w") - with self.test_session(): + with self.cached_session(): self.assertEqual(False, variables.is_variable_initialized(v).eval()) self.assertEqual(False, variables.is_variable_initialized(w).eval()) sm2 = session_manager.SessionManager( @@ -294,7 +294,7 @@ class SessionManagerTest(test.TestCase): trainable=False, collections=[ops.GraphKeys.LOCAL_VARIABLES], name="w") - with self.test_session(): + with self.cached_session(): self.assertEqual(False, variables.is_variable_initialized(v).eval()) self.assertEqual(False, variables.is_variable_initialized(w).eval()) sm2 = session_manager.SessionManager( @@ -326,7 +326,7 @@ class SessionManagerTest(test.TestCase): trainable=False, collections=[ops.GraphKeys.LOCAL_VARIABLES], name="w") - with self.test_session(): + with self.cached_session(): self.assertEqual(False, variables.is_variable_initialized(w).eval()) sm2 = session_manager.SessionManager( ready_op=variables.report_uninitialized_variables(), @@ -362,7 +362,7 @@ class SessionManagerTest(test.TestCase): trainable=False, collections=[ops.GraphKeys.LOCAL_VARIABLES], name="w") - with self.test_session(): + with self.cached_session(): self.assertEqual(False, variables.is_variable_initialized(v).eval()) self.assertEqual(False, variables.is_variable_initialized(w).eval()) sm2 = session_manager.SessionManager( @@ -467,7 +467,7 @@ class SessionManagerTest(test.TestCase): trainable=False, collections=[ops.GraphKeys.LOCAL_VARIABLES], name="x") - with self.test_session(): + with self.cached_session(): self.assertEqual(False, variables.is_variable_initialized(v).eval()) self.assertEqual(False, variables.is_variable_initialized(w).eval()) self.assertEqual(False, variables.is_variable_initialized(x).eval()) @@ -519,7 +519,7 @@ class SessionManagerTest(test.TestCase): collections=[ops.GraphKeys.LOCAL_VARIABLES], name="x_res") - with self.test_session(): + with self.cached_session(): self.assertEqual(False, variables.is_variable_initialized(v).eval()) self.assertEqual(False, variables.is_variable_initialized(w).eval()) self.assertEqual(False, variables.is_variable_initialized(x).eval()) @@ -566,7 +566,7 @@ class SessionManagerTest(test.TestCase): with ops.Graph().as_default(): i = control_flow_ops.while_loop(lambda i: i < 1, lambda i: i + 1, [0]) v = variables.Variable(array_ops.identity(i), name="v") - with self.test_session(): + with self.cached_session(): self.assertEqual(False, variables.is_variable_initialized(v).eval()) sm = session_manager.SessionManager( ready_op=variables.report_uninitialized_variables()) @@ -585,7 +585,7 @@ class SessionManagerTest(test.TestCase): trainable=False, collections=[ops.GraphKeys.LOCAL_VARIABLES], name="w") - with self.test_session(): + with self.cached_session(): self.assertEqual(False, variables.is_variable_initialized(v).eval()) self.assertEqual(False, variables.is_variable_initialized(w).eval()) sm2 = session_manager.SessionManager( @@ -602,7 +602,7 @@ class SessionManagerTest(test.TestCase): trainable=False, collections=[ops.GraphKeys.LOCAL_VARIABLES], name="w") - with self.test_session(): + with self.cached_session(): self.assertEqual(False, variables.is_variable_initialized(v).eval()) self.assertEqual(False, variables.is_variable_initialized(w).eval()) sm2 = session_manager.SessionManager( @@ -619,7 +619,7 @@ class SessionManagerTest(test.TestCase): trainable=False, collections=[ops.GraphKeys.LOCAL_VARIABLES], name="w") - with self.test_session(): + with self.cached_session(): self.assertEqual(False, variables.is_variable_initialized(v).eval()) self.assertEqual(False, variables.is_variable_initialized(w).eval()) sm2 = session_manager.SessionManager( @@ -640,7 +640,7 @@ class SessionManagerTest(test.TestCase): trainable=False, collections=[ops.GraphKeys.LOCAL_VARIABLES], name="w") - with self.test_session(): + with self.cached_session(): self.assertEqual(False, variables.is_variable_initialized(v).eval()) self.assertEqual(False, variables.is_variable_initialized(w).eval()) sm2 = session_manager.SessionManager( @@ -714,7 +714,7 @@ class ObsoleteSessionManagerTest(test.TestCase): os.rename(checkpoint_dir, checkpoint_dir2) gfile.MakeDirs(checkpoint_dir) v = variables.Variable([6.0, 7.0, 8.0], name="v") - with self.test_session(): + with self.cached_session(): self.assertEqual(False, variables.is_variable_initialized(v).eval()) session_manager.SessionManager( ready_op=variables.assert_variables_initialized()) @@ -769,7 +769,7 @@ class ObsoleteSessionManagerTest(test.TestCase): # Create a new Graph and SessionManager and recover. with ops.Graph().as_default(): v = variables.Variable(2, name="v") - with self.test_session(): + with self.cached_session(): self.assertEqual(False, variables.is_variable_initialized(v).eval()) sm2 = session_manager.SessionManager( ready_op=variables.assert_variables_initialized()) |