From dc0d78631cc08a91957619d3501b721b26f004c2 Mon Sep 17 00:00:00 2001 From: Ricardo Vieira Date: Fri, 17 Mar 2023 11:22:50 +0100 Subject: [PATCH] Fix JAX erfinv failing test JAX behavior at the boundary changed and no longer returns np.inf like scipy does. Fixed by evaluating at a non-boundary value. --- tests/link/jax/test_scalar.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/link/jax/test_scalar.py b/tests/link/jax/test_scalar.py index 235c971e39..260f198944 100644 --- a/tests/link/jax/test_scalar.py +++ b/tests/link/jax/test_scalar.py @@ -111,7 +111,7 @@ def test_erfinv(): out = erfinv(x) fg = FunctionGraph([x], [out]) - compare_jax_and_py(fg, [1.0]) + compare_jax_and_py(fg, [0.95]) def test_psi():