ENH: add atleast_nd

data-apis · Sep 20, 2024 · 60efb68 · 60efb68
1 parent da635ca
commit 60efb68
Show file tree

Hide file tree

Showing 6 changed files with 130 additions and 3 deletions.
diff --git a/pixi.lock b/pixi.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -28,16 +28,20 @@ classifiers = [
     "Typing :: Typed",
 ]
 dynamic = ["version"]
-dependencies = []
+dependencies = [
+    "array-api-compat",
+]
 
 [project.optional-dependencies]
 test = [
     "pytest >=6",
     "pytest-cov >=3",
+    "array-api-strict",
 ]
 dev = [
     "pytest >=6",
     "pytest-cov >=3",
+    "array-api-strict",
     "ruff",
     "pylint",
 ]

diff --git a/src/array_api_extra/__init__.py b/src/array_api_extra/__init__.py
@@ -1,5 +1,7 @@
 from __future__ import annotations
 
+from ._funcs import atleast_nd
+
 __version__ = "0.1.dev0"
 
-__all__ = ["__version__"]
+__all__ = ["__version__", "atleast_nd"]