some fixes

data-apis · Jan 11, 2024 · 762a03c · 762a03c
1 parent 64d20c4
commit 762a03c
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/array_api_compat/common/_aliases.py b/array_api_compat/common/_aliases.py
@@ -303,7 +303,7 @@ def _asarray(
         import numpy as xp
     elif namespace == 'cupy':
         import cupy as xp
-    elif namespace == 'dask':
+    elif namespace == 'dask.array':
         import dask.array as xp
     else:
         raise ValueError("Unrecognized namespace argument to asarray()")

diff --git a/array_api_compat/dask/array/_aliases.py b/array_api_compat/dask/array/_aliases.py
@@ -76,7 +76,7 @@ def dask_arange(
 eye = get_xp(da)(_aliases.eye)
 
 from functools import partial
-asarray = partial(_aliases._asarray, namespace='dask')
+asarray = partial(_aliases._asarray, namespace='dask.array')
 asarray.__doc__ = _aliases._asarray.__doc__
 
 linspace = get_xp(da)(_aliases.linspace)

diff --git a/tests/test_array_namespace.py b/tests/test_array_namespace.py
@@ -17,8 +17,9 @@ def test_array_namespace(library, api_version):
         assert namespace == lib
     else:
         if library == "dask.array":
-            library = "dask"
-        assert namespace == getattr(array_api_compat, library)
+            assert namespace == array_api_compat.dask.array
+        else:
+            assert namespace == getattr(array_api_compat, library)
 
 
 def test_array_namespace_errors():