diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-08-21 19:24:19 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-21 19:27:54 -0700 |
| commit | 496023e9dc84a076caeb2e5e8e13b6a3d819ad6d (patch) | |
| tree | 9776c9865f7b98a15817bc6be4c2b683323a67b1 /tensorflow/contrib/tensor_forest | |
| parent | 361a82d73a50a800510674b3aaa20e4845e56434 (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: 209701635
Diffstat (limited to 'tensorflow/contrib/tensor_forest')
2 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/tensor_forest/hybrid/python/kernel_tests/k_feature_routing_function_op_test.py b/tensorflow/contrib/tensor_forest/hybrid/python/kernel_tests/k_feature_routing_function_op_test.py index 980f53253d..cc053f3b94 100644 --- a/tensorflow/contrib/tensor_forest/hybrid/python/kernel_tests/k_feature_routing_function_op_test.py +++ b/tensorflow/contrib/tensor_forest/hybrid/python/kernel_tests/k_feature_routing_function_op_test.py @@ -58,7 +58,7 @@ class KFeatureRoutingFunctionTest(test_util.TensorFlowTestCase): self.assertEquals(self.params.num_features_per_node, 2) def testRoutingFunction(self): - with self.test_session(): + with self.cached_session(): route_tensor = gen_training_ops.k_feature_routing_function( self.input_data, self.tree_weights, diff --git a/tensorflow/contrib/tensor_forest/hybrid/python/kernel_tests/routing_function_op_test.py b/tensorflow/contrib/tensor_forest/hybrid/python/kernel_tests/routing_function_op_test.py index a27fd49d32..554f7b0d7a 100644 --- a/tensorflow/contrib/tensor_forest/hybrid/python/kernel_tests/routing_function_op_test.py +++ b/tensorflow/contrib/tensor_forest/hybrid/python/kernel_tests/routing_function_op_test.py @@ -36,7 +36,7 @@ class RoutingFunctionTest(test_util.TensorFlowTestCase): self.ops = training_ops.Load() def testRoutingFunction(self): - with self.test_session(): + with self.cached_session(): route_tensor = gen_training_ops.routing_function( self.input_data, self.tree_weights, self.tree_thresholds, max_nodes=3) |