diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-10 14:37:06 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-10 15:04:14 -0700 |
| commit | b828f89263e054bfa7c7a808cab1506834ab906d (patch) | |
| tree | e31816a6850d177306f19ee8670e0836060fcfc9 /tensorflow/python/training | |
| parent | acf0ee82092727afc2067316982407cf5e496f75 (diff) | |
Move from deprecated self.test_session() to self.cached_session().
self.test_session() has been deprecated in 9962eb5e84b15e309410071b06c2ed2d6148ed44 as its name confuses readers of the test. Moving to cached_session() instead which is more explicit about:
* the fact that the session may be reused.
* the session is not closed even when doing a "with self.test_session()" statement.
PiperOrigin-RevId: 212336464
Diffstat (limited to 'tensorflow/python/training')
| -rw-r--r-- | tensorflow/python/training/checkpointable/tracking_test.py | 2 | ||||
| -rw-r--r-- | tensorflow/python/training/checkpointable/util_test.py | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/training/checkpointable/tracking_test.py b/tensorflow/python/training/checkpointable/tracking_test.py index e85f812ce2..a44c570fb9 100644 --- a/tensorflow/python/training/checkpointable/tracking_test.py +++ b/tensorflow/python/training/checkpointable/tracking_test.py @@ -165,7 +165,7 @@ class InterfaceTests(test.TestCase): self.assertEqual([c], a.attribute["c"].layers) checkpoint = util.Checkpoint(a=a) save_path = checkpoint.save(os.path.join(self.get_temp_dir(), "ckpt")) - with self.test_session(): + with self.cached_session(): checkpoint.restore(save_path).assert_consumed().initialize_or_restore() @test_util.run_in_graph_and_eager_modes diff --git a/tensorflow/python/training/checkpointable/util_test.py b/tensorflow/python/training/checkpointable/util_test.py index 0d32d21426..f8b5bd8501 100644 --- a/tensorflow/python/training/checkpointable/util_test.py +++ b/tensorflow/python/training/checkpointable/util_test.py @@ -384,7 +384,7 @@ class CheckpointingTests(test.TestCase): saver = saver_lib.Saver(var_list=[v]) test_dir = self.get_temp_dir() prefix = os.path.join(test_dir, "ckpt") - with self.test_session() as sess: + with self.cached_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.)) |