From 4066a2adfddf659acc268c123d121e23a09d8537 Mon Sep 17 00:00:00 2001 From: leben Date: Fri, 16 Aug 2024 05:33:41 -0700 Subject: [PATCH] Increase `test_make_lanczos_preconditioner` atol to 0.15. PiperOrigin-RevId: 663699118 --- .../python/experimental/fastgp/partial_lanczos_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow_probability/python/experimental/fastgp/partial_lanczos_test.py b/tensorflow_probability/python/experimental/fastgp/partial_lanczos_test.py index 89d4b8204c..aa12ffa2ba 100644 --- a/tensorflow_probability/python/experimental/fastgp/partial_lanczos_test.py +++ b/tensorflow_probability/python/experimental/fastgp/partial_lanczos_test.py @@ -128,7 +128,7 @@ def test_make_lanczos_preconditioner(self): log_det = preconditioner.log_abs_determinant() self.assertAlmostEqual(0.0, log_det, places=4) out = preconditioner.solve(jnp.identity(10)) - np.testing.assert_allclose(out, jnp.identity(10), atol=9e-2) + np.testing.assert_allclose(out, jnp.identity(10), atol=0.15) kernel = jnp.identity(100).astype(self.dtype) preconditioner = partial_lanczos.make_lanczos_preconditioner( kernel, jax.random.PRNGKey(6)