Minor (slightly increase test tolerances)

ltatzel · Jun 14, 2024 · 32e399c · 32e399c
1 parent d4e8185
commit 32e399c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_cg.py b/tests/test_cg.py
@@ -28,7 +28,7 @@
 
 # The "incrementally" computed residual in cg might differ slightly from
 # `A @ x_cg - b`.
-EPS = 5e-7
+EPS = 5e-6
 
 
 @pytest.mark.parametrize("seed", SEEDS, ids=SEEDS_IDS)
@@ -153,7 +153,7 @@ def quadratic(x):
     m_vals = torch.Tensor(m_iters)
 
     error_msg = "Discrepancy between quadratic and `m_iters`"
-    assert torch.allclose(quadratic_vals, m_vals), error_msg
+    assert torch.allclose(quadratic_vals, m_vals, atol=1e-7), error_msg
 
 
 @pytest.mark.parametrize("seed", SEEDS, ids=SEEDS_IDS)