changes left behind

data-apis · Jan 9, 2024 · df69086 · df69086
1 parent 0d66160
commit df69086
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/array_api_compat/common/_helpers.py b/array_api_compat/common/_helpers.py
@@ -110,7 +110,7 @@ def your_function(x, y):
         elif _is_dask_array(x):
             _check_api_version(api_version)
             if _use_compat:
-                from .. import dask as dask_namespace
+                from ..dask import array as dask_namespace
                 namespaces.add(dask_namespace)
             else:
                 raise TypeError("_use_compat cannot be False if input array is a dask array!")

diff --git a/array_api_compat/dask/array/linalg.py b/array_api_compat/dask/array/linalg.py
@@ -1,8 +1,8 @@
 from dask.array.linalg import *
 from dask.array.linalg import __all__ as linalg_all
 
-from ..common import _linalg
-from .._internal import get_xp
+from ...common import _linalg
+from ..._internal import get_xp
 from ._aliases import (matmul, matrix_transpose, tensordot, vecdot)
 
 import dask.array as da