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 d091f79Copy full SHA for d091f79
array_api_compat/common/_helpers.py
@@ -521,11 +521,15 @@ 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)
- else:
531
+ elif _use_compat is False:
532
+ print("why am i false")
533
import dask.array as da
534
namespaces.add(da)
535
elif is_jax_array(x):
0 commit comments