API: remove dependency on array-api-compat

Author: lucascolley

pixi.lock

@@ -28,9 +28,7 @@ classifiers = [
     "Typing :: Typed",
 ]
 dynamic = ["version"]
-dependencies = [
-    "array-api-compat",
-]
+dependencies = []
 
 [project.optional-dependencies]
 test = [

pyproject.toml

@@ -2,15 +2,13 @@
 
 from typing import TYPE_CHECKING
 
-from array_api_compat import array_namespace  # type: ignore[import-not-found]
-
 if TYPE_CHECKING:
     from ._typing import Array, ModuleType
 
 __all__ = ["atleast_nd"]
 
 
-def atleast_nd(x: Array, *, ndim: int, xp: ModuleType | None = None) -> Array:
+def atleast_nd(x: Array, *, ndim: int, xp: ModuleType) -> Array:
     """
     Recursively expand the dimension of an array to have at least `ndim`.
 
@@ -19,15 +17,19 @@ def atleast_nd(x: Array, *, ndim: int, xp: ModuleType | None = None) -> Array:
     x: array
         An array.
 
+    ndim: int
+        The minimum number of dimensions for the result.
+
+    xp: array_namespace
+        The array namespace for `x`.
+
     Returns
     -------
     res: array
         An array with ``res.ndim`` >= `ndim`.
         If ``x.ndim`` >= `ndim`, `x` is returned.
         If ``x.ndim`` < `ndim`, ``res.ndim`` will equal `ndim`.
     """
-    xp = array_namespace(x) if xp is None else xp
-
     x = xp.asarray(x)
     if x.ndim < ndim:
         x = xp.expand_dims(x, axis=0)