We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent cdd1c8d commit 6b11b2bCopy full SHA for 6b11b2b
array_api_compat/common/_helpers.py
@@ -521,7 +521,10 @@ def your_function(x, y):
521
import torch
522
namespaces.add(torch)
523
elif is_dask_array(x):
524
- if _use_compat:
+ # dask main namespace is not array API compatible
525
+ # so return namespace from array-api-compat unless
526
+ # explicitly requested otherwise
527
+ if _use_compat or _use_compat is None:
528
_check_api_version(api_version)
529
from ..dask import array as dask_namespace
530
namespaces.add(dask_namespace)
0 commit comments