diff options
Diffstat (limited to 'tensorflow/contrib/solvers/python/kernel_tests/least_squares_test.py')
-rw-r--r-- | tensorflow/contrib/solvers/python/kernel_tests/least_squares_test.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/solvers/python/kernel_tests/least_squares_test.py b/tensorflow/contrib/solvers/python/kernel_tests/least_squares_test.py index a73642716b..2a9100903a 100644 --- a/tensorflow/contrib/solvers/python/kernel_tests/least_squares_test.py +++ b/tensorflow/contrib/solvers/python/kernel_tests/least_squares_test.py @@ -47,7 +47,7 @@ def _get_least_squares_tests(dtype_, use_static_shape_, shape_): low=-1.0, high=1.0, size=shape_[0]).astype(dtype_) tol = 1e-12 if dtype_ == np.float64 else 1e-6 max_iter = 20 - with self.test_session() as sess: + with self.cached_session() as sess: if use_static_shape_: a = constant_op.constant(a_np) rhs = constant_op.constant(rhs_np) |