提交 651ed9f4 authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Thomas Wiecki

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.
上级 277559b4
...@@ -111,7 +111,7 @@ def test_erfinv(): ...@@ -111,7 +111,7 @@ def test_erfinv():
out = erfinv(x) out = erfinv(x)
fg = FunctionGraph([x], [out]) fg = FunctionGraph([x], [out])
compare_jax_and_py(fg, [1.0]) compare_jax_and_py(fg, [0.95])
def test_psi(): def test_psi():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论