diff --git a/test/test_re/test_evi.py b/test/test_re/test_evi.py index b4f990342030792e213cb8c72a08b5f3943db539..2f65616ddae6b61cef6023a5f8363aedcc239d8f 100644 --- a/test/test_re/test_evi.py +++ b/test/test_re/test_evi.py @@ -198,7 +198,7 @@ def test_mgvi_wiener_filter_consistency( odir=None, ) approx_post_mean = jax.vmap(forward)(samples_opt.samples).mean(axis=0) - assert_allclose(post_mean, approx_post_mean, atol=1e-10, rtol=1e-10) + assert_allclose(post_mean, approx_post_mean, atol=1e-9, rtol=1e-10) # Wiener filter test key, w_key = random.split(key) diff --git a/test/test_re/test_likelihood_impl.py b/test/test_re/test_likelihood_impl.py index c251e4679265143bddef97fc0164cd35cc5361c6..a531305324604284c5a78560dbba773787c999eb 100644 --- a/test/test_re/test_likelihood_impl.py +++ b/test/test_re/test_likelihood_impl.py @@ -149,7 +149,7 @@ lh_init_approx = ( def test_gaussian_vs_vcgaussian_consistency(seed, shape): rtol = 10 * jnp.finfo(jnp.zeros(0).dtype).eps - atol = 5 * jnp.finfo(jnp.zeros(0).dtype).eps + atol = 10 * jnp.finfo(jnp.zeros(0).dtype).eps key = random.PRNGKey(seed) sk = list(random.split(key, 5))