WIP MAINT: upgrade to array-api-compat >=1.12

Author: crusaderky

pyproject.toml

@@ -26,7 +26,7 @@ classifiers = [
   "Typing :: Typed",
 ]
 dynamic = ["version"]
-dependencies = ["array-api-compat>=1.11.2,<2"]
+dependencies = ["array-api-compat>=1.12.0,<2"]
 
 [project.urls]
 Homepage = "https://github.com/data-apis/array-api-extra"
@@ -48,7 +48,7 @@ platforms = ["linux-64", "osx-64", "osx-arm64", "win-64"]
 
 [tool.pixi.dependencies]
 python = ">=3.10,<3.14"
-array-api-compat = ">=1.11.2,<2"
+array-api-compat = ">=1.12.0,<2"
 
 [tool.pixi.pypi-dependencies]
 array-api-extra = { path = ".", editable = true }

src/array_api_extra/_lib/_utils/_helpers.py

@@ -296,13 +296,9 @@ def capabilities(xp: ModuleType) -> dict[str, int]:
         # No __array_namespace_info__(); no indexing by sparse arrays
         return {"boolean indexing": False, "data-dependent shapes": True}
     out = xp.__array_namespace_info__().capabilities()
-    if is_jax_namespace(xp):
+    if is_jax_namespace(xp) and out["boolean indexing"]:
         # FIXME https://github.com/jax-ml/jax/issues/27418
+        # Fixed in jax >=0.6.0
         out = out.copy()
         out["boolean indexing"] = False
-    if is_dask_namespace(xp):
-        # FIXME https://github.com/data-apis/array-api-compat/pull/290
-        out = out.copy()
-        out["boolean indexing"] = True
-        out["data-dependent shapes"] = True
     return out