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 5affae5 commit c6cce57Copy full SHA for c6cce57
array_api_compat/common/_helpers.py
@@ -519,11 +519,15 @@ def your_function(x, y):
519
import torch
520
namespaces.add(torch)
521
elif is_dask_array(x):
522
- if _use_compat:
+ # dask main namespace is not array APi compatible
523
+ # so return namespace from array-api-compat unless
524
+ # explicitly requested otherwise
525
+ if _use_compat or _use_compat is None:
526
_check_api_version(api_version)
527
from ..dask import array as dask_namespace
528
namespaces.add(dask_namespace)
- else:
529
+ elif _use_compat is False:
530
+ print("why am i false")
531
import dask.array as da
532
namespaces.add(da)
533
elif is_jax_array(x):
0 commit comments