diff options
Diffstat (limited to 'tensorflow/python/kernel_tests/sparsemask_op_test.py')
-rw-r--r-- | tensorflow/python/kernel_tests/sparsemask_op_test.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/kernel_tests/sparsemask_op_test.py b/tensorflow/python/kernel_tests/sparsemask_op_test.py index cf6c9494ae..6f5dd45b61 100644 --- a/tensorflow/python/kernel_tests/sparsemask_op_test.py +++ b/tensorflow/python/kernel_tests/sparsemask_op_test.py @@ -34,7 +34,7 @@ class SparseMaskTest(test.TestCase): out_values = values[1:, :] out_indices = np.array([2, 3, 4], dtype=np.int32) - with self.test_session() as sess: + with self.cached_session() as sess: values_tensor = ops.convert_to_tensor(values) indices_tensor = ops.convert_to_tensor(indices) mask_indices_tensor = ops.convert_to_tensor(mask_indices) |