Merge pull request #26542 from jakevdp:fix-breakage

PiperOrigin-RevId: 726944183
jax-ml · Feb 14, 2025 · ca87f5f · ca87f5f
2 parents 6addf02 + b93934c
commit ca87f5f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jax/_src/numpy/indexing.py b/jax/_src/numpy/indexing.py
@@ -1003,7 +1003,7 @@ def _expand_bool_indices(idx, shape):
       abstract_i = None
     if _is_boolean_index(i):
       if isinstance(i, list):
-        i = array(i)
+        i = lax_numpy.array(i)
         abstract_i = core.get_aval(i)
 
       if not core.is_concrete(i):