Fix failure

data-apis · Jan 7, 2025 · 82fab45 · 82fab45
1 parent 1b89e4d
commit 82fab45
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/test_common.py b/tests/test_common.py
@@ -116,6 +116,11 @@ def test_is_lazy_array_unknown(array, monkeypatch):
         "is_jax_array", 
         lambda x: False,
     )
+    monkeypatch.setattr(  # Needed for JAX < 0.4.32
+        array_api_compat.common._helpers, 
+        "array_namespace", 
+        lambda x: xp,
+    )
 
     assert not is_lazy_array(x)  # Eager JAX
     assert jax.jit(is_lazy_array)(x)  # Jitted (lazy) JAX