Skip to content

Commit 2b5fcc5

Browse files
committed
Ignore JAX bug
1 parent ad8c777 commit 2b5fcc5

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/test_funcs.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -258,7 +258,10 @@ def f2(*args: Array) -> Array:
258258
# TODO remove asarrays once all backends support Array API 2024.12
259259
ref3 = xp.where(cond, *asarrays(f1(*arrays), float_fill_value, xp=xp))
260260

261-
xp_assert_close(res1, ref1, rtol=2e-16)
261+
# https://github.com/jax-ml/jax/issues/26658
262+
atol = 1e-300 if library is Backend.JAX else 0
263+
264+
xp_assert_close(res1, ref1, atol=atol, rtol=2e-16)
262265
xp_assert_equal(res2, ref2)
263266
xp_assert_equal(res3, ref3)
264267

0 commit comments

Comments
 (0)