diff options
author | 2018-08-23 12:53:57 -0700 | |
---|---|---|
committer | 2018-08-23 12:57:31 -0700 | |
commit | eabacb190e6cdfa2785c75924eca5f5b5401c24c (patch) | |
tree | 8a1ae006ccf1a46af93006becf83f78c243c76dc /tensorflow/python/training | |
parent | cddc30aaafc35868848608cd4db40b73ff42bbfe (diff) |
Fix checkpointable/util_test and training/saver_test.
The evaluate should be within the session scope.
PiperOrigin-RevId: 209983400
Diffstat (limited to 'tensorflow/python/training')
-rw-r--r-- | tensorflow/python/training/checkpointable/util_test.py | 2 | ||||
-rw-r--r-- | tensorflow/python/training/saver_test.py | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/training/checkpointable/util_test.py b/tensorflow/python/training/checkpointable/util_test.py index ecadc56871..697b44c3ff 100644 --- a/tensorflow/python/training/checkpointable/util_test.py +++ b/tensorflow/python/training/checkpointable/util_test.py @@ -384,8 +384,8 @@ class CheckpointingTests(test.TestCase): saver = saver_lib.Saver(var_list=[v]) test_dir = self.get_temp_dir() prefix = os.path.join(test_dir, "ckpt") - self.evaluate(v.non_dep_variable.assign(42.)) with self.test_session() as sess: + self.evaluate(v.non_dep_variable.assign(42.)) save_path = saver.save(sess, prefix) self.evaluate(v.non_dep_variable.assign(43.)) self.evaluate(v.mirrored.assign(44.)) diff --git a/tensorflow/python/training/saver_test.py b/tensorflow/python/training/saver_test.py index b46095d458..f5b2a22327 100644 --- a/tensorflow/python/training/saver_test.py +++ b/tensorflow/python/training/saver_test.py @@ -2853,8 +2853,8 @@ class CheckpointableCompatibilityTests(test.TestCase): saver = saver_module.Saver(var_list=[v]) test_dir = self.get_temp_dir() prefix = os.path.join(test_dir, "ckpt") - self.evaluate(v.non_dep_variable.assign(42.)) with self.test_session() as sess: + self.evaluate(v.non_dep_variable.assign(42.)) save_path = saver.save(sess, prefix) self.evaluate(v.non_dep_variable.assign(43.)) saver.restore(sess, save_path) |