Wrap dask.array by default since it is not Array API Compatible

data-apis · Dec 25, 2024 · 6b11b2b · 6b11b2b
1 parent cdd1c8d
commit 6b11b2b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/array_api_compat/common/_helpers.py b/array_api_compat/common/_helpers.py
@@ -521,7 +521,10 @@ def your_function(x, y):
                 import torch
                 namespaces.add(torch)
         elif is_dask_array(x):
-            if _use_compat:
+            # dask main namespace is not array API compatible
+            # so return namespace from array-api-compat unless
+            # explicitly requested otherwise
+            if _use_compat or _use_compat is None:
                 _check_api_version(api_version)
                 from ..dask import array as dask_namespace
                 namespaces.add(dask_namespace)