From 85d49828354a017b270cb0555ce87b878091ece4 Mon Sep 17 00:00:00 2001 From: Qazalbash Date: Mon, 27 Jan 2025 23:56:17 +0500 Subject: [PATCH] fix(tests): increase relative tolerance for `test_cholesky_update` --- test/test_distributions_util.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_distributions_util.py b/test/test_distributions_util.py index 2a652fd46..bc82854d2 100644 --- 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])