Loosen amount (correct test)

Ch0ronomato · Ch0ronomato · commit f3578e62a8f2 · 2024-11-01T22:14:19.000-07:00
diff --git a/tests/tensor/test_blockwise.py b/tests/tensor/test_blockwise.py
@@ -484,7 +484,7 @@ def test_cholesky(self, is_batched):
         np.testing.assert_allclose(
             X_val_in,
             np.linalg.cholesky(X_val_in_copy),
-            atol=1e-4 if config.floatX == "float32" else 0,
+            atol=1e-5 if config.floatX == "float32" else 0,
         )
 
     @pytest.mark.parametrize("batched_A", (False, True))
@@ -538,7 +538,7 @@ def core_scipy_fn(A, b):
             A_val_copy, b_val_copy
         )
         np.testing.assert_allclose(
-            out, expected_out, atol=1e-6 if config.floatX == "float32" else 0
+            out, expected_out, atol=1e-5 if config.floatX == "float32" else 0
         )
 
         # Confirm input was destroyed

Original file line number	Diff line number	Diff line change
`@@ -484,7 +484,7 @@ def test_cholesky(self, is_batched):`
`484`	`484`	`np.testing.assert_allclose(`
`485`	`485`	`X_val_in,`
`486`	`486`	`np.linalg.cholesky(X_val_in_copy),`
`487`		`- atol=1e-4 if config.floatX == "float32" else 0,`
	`487`	`+ atol=1e-5 if config.floatX == "float32" else 0,`
`488`	`488`	`)`
`489`	`489`
`490`	`490`	`@pytest.mark.parametrize("batched_A", (False, True))`
`@@ -538,7 +538,7 @@ def core_scipy_fn(A, b):`
`538`	`538`	`A_val_copy, b_val_copy`
`539`	`539`	`)`
`540`	`540`	`np.testing.assert_allclose(`
`541`		`- out, expected_out, atol=1e-6 if config.floatX == "float32" else 0`
	`541`	`+ out, expected_out, atol=1e-5 if config.floatX == "float32" else 0`
`542`	`542`	`)`
`543`	`543`
`544`	`544`	`# Confirm input was destroyed`