diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-21 00:02:49 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-21 00:07:39 -0700 |
commit | 2952f5134905af795ba90ae1eb97e39091ba9843 (patch) | |
tree | f73bc5cd0342d9449114bd933863c2aa55610aa2 /tensorflow/contrib/tensor_forest/python | |
parent | cf047f7755f3400ee128db2571042091fe9f8314 (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: 213944355
Diffstat (limited to 'tensorflow/contrib/tensor_forest/python')
-rw-r--r-- | tensorflow/contrib/tensor_forest/python/kernel_tests/scatter_add_ndim_op_test.py | 10 | ||||
-rw-r--r-- | tensorflow/contrib/tensor_forest/python/tensor_forest_test.py | 2 |
2 files changed, 6 insertions, 6 deletions
diff --git a/tensorflow/contrib/tensor_forest/python/kernel_tests/scatter_add_ndim_op_test.py b/tensorflow/contrib/tensor_forest/python/kernel_tests/scatter_add_ndim_op_test.py index e429d12e96..1c4e18dbda 100644 --- a/tensorflow/contrib/tensor_forest/python/kernel_tests/scatter_add_ndim_op_test.py +++ b/tensorflow/contrib/tensor_forest/python/kernel_tests/scatter_add_ndim_op_test.py @@ -32,7 +32,7 @@ class ScatterAddNdimTest(test_util.TensorFlowTestCase): indices = [[1], [10]] updates = [100., 200.] - with self.test_session(): + with self.cached_session(): variables.global_variables_initializer().run() tensor_forest_ops.scatter_add_ndim(input_data, indices, updates).run() self.assertAllEqual( @@ -45,7 +45,7 @@ class ScatterAddNdimTest(test_util.TensorFlowTestCase): indices = [[0, 0, 1], [1, 1, 2]] updates = [100., 200.] - with self.test_session(): + with self.cached_session(): variables.global_variables_initializer().run() tensor_forest_ops.scatter_add_ndim(input_data, indices, updates).run() self.assertAllEqual([[[1., 102., 3.], [4., 5., 6.]], @@ -57,7 +57,7 @@ class ScatterAddNdimTest(test_util.TensorFlowTestCase): indices = [] updates = [] - with self.test_session(): + with self.cached_session(): variables.global_variables_initializer().run() tensor_forest_ops.scatter_add_ndim(input_data, indices, updates).run() self.assertAllEqual(init_val, input_data.eval()) @@ -67,7 +67,7 @@ class ScatterAddNdimTest(test_util.TensorFlowTestCase): input_data = variables.Variable(init_val) indices = [[0, 0, 1], [1, 1, 2]] updates = [100.] - with self.test_session(): + with self.cached_session(): variables.global_variables_initializer().run() with self.assertRaisesOpError( 'Number of updates should be same as number of indices.'): @@ -80,7 +80,7 @@ class ScatterAddNdimTest(test_util.TensorFlowTestCase): indices = [[0, 0], [1, 1]] updates = [[100., 200., 300.], [400., 500., 600.]] - with self.test_session(): + with self.cached_session(): variables.global_variables_initializer().run() tensor_forest_ops.scatter_add_ndim(input_data, indices, updates).run() self.assertAllEqual([[[101., 202., 303.], [4., 5., 6.]], diff --git a/tensorflow/contrib/tensor_forest/python/tensor_forest_test.py b/tensorflow/contrib/tensor_forest/python/tensor_forest_test.py index 1c9c81827e..e0f0c0d4ff 100644 --- a/tensorflow/contrib/tensor_forest/python/tensor_forest_test.py +++ b/tensorflow/contrib/tensor_forest/python/tensor_forest_test.py @@ -149,7 +149,7 @@ class TensorForestTest(test_util.TensorFlowTestCase): self.assertTrue(isinstance(probs, ops.Tensor)) self.assertTrue(isinstance(paths, ops.Tensor)) self.assertTrue(isinstance(var, ops.Tensor)) - with self.test_session(): + with self.cached_session(): variables.global_variables_initializer().run() resources.initialize_resources(resources.shared_resources()).run() self.assertEquals(probs.eval().shape, (4, 2)) |