Skip to content

Commit 762a03c

Browse files
committed
some fixes
1 parent 64d20c4 commit 762a03c

File tree

3 files changed

+5
-4
lines changed

3 files changed

+5
-4
lines changed

array_api_compat/common/_aliases.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -303,7 +303,7 @@ def _asarray(
303303
import numpy as xp
304304
elif namespace == 'cupy':
305305
import cupy as xp
306-
elif namespace == 'dask':
306+
elif namespace == 'dask.array':
307307
import dask.array as xp
308308
else:
309309
raise ValueError("Unrecognized namespace argument to asarray()")

array_api_compat/dask/array/_aliases.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ def dask_arange(
7676
eye = get_xp(da)(_aliases.eye)
7777

7878
from functools import partial
79-
asarray = partial(_aliases._asarray, namespace='dask')
79+
asarray = partial(_aliases._asarray, namespace='dask.array')
8080
asarray.__doc__ = _aliases._asarray.__doc__
8181

8282
linspace = get_xp(da)(_aliases.linspace)

tests/test_array_namespace.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,9 @@ def test_array_namespace(library, api_version):
1717
assert namespace == lib
1818
else:
1919
if library == "dask.array":
20-
library = "dask"
21-
assert namespace == getattr(array_api_compat, library)
20+
assert namespace == array_api_compat.dask.array
21+
else:
22+
assert namespace == getattr(array_api_compat, library)
2223

2324

2425
def test_array_namespace_errors():

0 commit comments

Comments
 (0)