fix(tests): increase relative tolerance for test_cholesky_update

pyro-ppl · Jan 27, 2025 · 85d4982 · 85d4982
1 parent 7c16f0c
commit 85d4982
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_distributions_util.py b/test/test_distributions_util.py
@@ -140,7 +140,7 @@ def test_cholesky_update(chol_batch_shape, vec_batch_shape, dim, coef):
     xxt = x[..., None] @ x[..., None, :]
     expected = jnp.linalg.cholesky(A + coef * xxt)
     actual = cholesky_update(jnp.linalg.cholesky(A), x, coef)
-    assert_allclose(actual, expected, atol=3.8e-4, rtol=1e-4)
+    assert_allclose(actual, expected, atol=3.8e-4, rtol=8e-4)
 
 
 @pytest.mark.parametrize("n", [10, 100, 1000])