lint

data-apis · Jan 9, 2025 · 72360ed · 72360ed
1 parent 7f66c3d
commit 72360ed
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/src/array_api_extra/_apply.py b/src/array_api_extra/_apply.py
@@ -8,21 +8,21 @@
 from types import ModuleType
 from typing import TYPE_CHECKING, Any, cast
 
+from ._lib._compat import (
+    array_namespace,
+    is_dask_namespace,
+    is_jax_namespace,
+)
+from ._lib._typing import Array, DType
+
 if TYPE_CHECKING:
     # https://github.com/scikit-learn/scikit-learn/pull/27910#issuecomment-2568023972
     from typing import TypeAlias
 
     import numpy as np
     import numpy.typing as npt
 
-    NumPyObject: TypeAlias = npt.NDArray[Any] | np.generic  # type: ignore[no-any-explicit]
-
-from ._lib._compat import (
-    array_namespace,
-    is_dask_namespace,
-    is_jax_namespace,
-)
-from ._lib._typing import Array, DType
+    NumPyObject: TypeAlias = npt.NDArray[DType] | np.generic  # type: ignore[no-any-explicit]
 
 
 def apply_numpy_func(  # type: ignore[no-any-explicit]