Skip to content

Commit 82fab45

Browse files
committed
Fix failure
1 parent 1b89e4d commit 82fab45

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

Diff for: tests/test_common.py

+5
Original file line numberDiff line numberDiff line change
@@ -116,6 +116,11 @@ def test_is_lazy_array_unknown(array, monkeypatch):
116116
"is_jax_array",
117117
lambda x: False,
118118
)
119+
monkeypatch.setattr( # Needed for JAX < 0.4.32
120+
array_api_compat.common._helpers,
121+
"array_namespace",
122+
lambda x: xp,
123+
)
119124

120125
assert not is_lazy_array(x) # Eager JAX
121126
assert jax.jit(is_lazy_array)(x) # Jitted (lazy) JAX

0 commit comments

Comments
 (0)