diff options
Diffstat (limited to 'tensorflow/python/kernel_tests/sparse_ops_test.py')
| -rw-r--r-- | tensorflow/python/kernel_tests/sparse_ops_test.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/kernel_tests/sparse_ops_test.py b/tensorflow/python/kernel_tests/sparse_ops_test.py index fc39de150e..79efee3f5b 100644 --- a/tensorflow/python/kernel_tests/sparse_ops_test.py +++ b/tensorflow/python/kernel_tests/sparse_ops_test.py @@ -628,7 +628,7 @@ class SparseReduceTest(test_util.TensorFlowTestCase): else: np_ans = np.max(np_ans, axis=ra, keepdims=keep_dims) - with self.test_session(): + with self.cached_session(): if do_sum: tf_dense_ans = sparse_ops.sparse_reduce_sum(sp_t, reduction_axes, keep_dims) |