Skip to content
Snippets Groups Projects
Commit 465c1141 authored by Jakob Roth's avatar Jakob Roth Committed by Gordian Edenhofer
Browse files

test_re: increase numerical tolerance

parent 7462d5f9
No related branches found
No related tags found
1 merge request!976Remove jax version constraint as recent version works again
......@@ -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)
......
......@@ -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))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment