No public description

PiperOrigin-RevId: 675606163
google · Sep 17, 2024 · 015acbd · 015acbd
1 parent 66d90db
commit 015acbd
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/linear_solve_test.py b/tests/linear_solve_test.py
@@ -128,11 +128,11 @@ def matvec(x):
     x5 = linear_solve.solve_iterative_refinement(matvec, b)
     x6 = linear_solve.solve_qr(matvec, b)
 
-    self.assertArraysAllClose(x, x2, atol=1e-4)
-    self.assertArraysAllClose(x, x3, atol=1e-4)
-    self.assertArraysAllClose(x, x4, atol=1e-4)
-    self.assertArraysAllClose(x, x5, atol=1e-4)
-    self.assertArraysAllClose(x, x6, atol=1e-4)
+    self.assertArraysAllClose(x, x2, atol=1e-4, rtol=1e-4)
+    self.assertArraysAllClose(x, x3, atol=1e-4, rtol=1e-4)
+    self.assertArraysAllClose(x, x4, atol=1e-4, rtol=1e-4)
+    self.assertArraysAllClose(x, x5, atol=1e-4, rtol=1e-4)
+    self.assertArraysAllClose(x, x6, atol=1e-4, rtol=1e-4)
 
   def test_solve_sparse_ridge(self):
     rng = onp.random.RandomState(0)