Increase fast MTGP test tolerance.

PiperOrigin-RevId: 607673435
tensorflow · Feb 16, 2024 · b24cdb2 · b24cdb2
1 parent c9a22d6
commit b24cdb2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow_probability/python/experimental/fastgp/fast_mtgp_test.py b/tensorflow_probability/python/experimental/fastgp/fast_mtgp_test.py
@@ -260,7 +260,7 @@ def test_gp_log_prob_hard(self):
     )
     fgp_lp = jnp.sum(fgp.log_prob(samples, key=jax.random.PRNGKey(1)))
     sgp_lp = jnp.sum(sgp.log_prob(samples))
-    np.testing.assert_allclose(fgp_lp, sgp_lp, rtol=3e-4)
+    np.testing.assert_allclose(fgp_lp, sgp_lp, rtol=4e-4)
 
   def test_gp_log_prob_matern_five_halves(self):
     kernel = tfp.math.psd_kernels.MaternFiveHalves(