array-api-extra 0.10.1__tar.gz → 0.10.3__tar.gz

{array_api_extra-0.10.1 → array_api_extra-0.10.3}/.all-contributorsrc

@@ -233,7 +233,8 @@
       "profile": "https://github.com/NeilGirdhar",
       "contributions": [
         "code",
-        "test"
+        "test",
+        "review"
       ]
     },
     {
@@ -319,6 +320,27 @@
         "ideas",
         "test"
       ]
+    },
+    {
+      "login": "prady0t",
+      "name": "Pradyot Ranjan",
+      "avatar_url": "https://avatars.githubusercontent.com/u/99216956?v=4",
+      "profile": "https://github.com/prady0t",
+      "contributions": [
+        "code",
+        "doc",
+        "ideas",
+        "test"
+      ]
+    },
+    {
+      "login": "lundybernard",
+      "name": "Lundy Bernard",
+      "avatar_url": "https://avatars.githubusercontent.com/u/17297695?v=4",
+      "profile": "https://github.com/lundybernard",
+      "contributions": [
+        "review"
+      ]
     }
   ]
 }

{array_api_extra-0.10.1 → array_api_extra-0.10.3}/CONTRIBUTORS.md

@@ -33,7 +33,7 @@ This project exists thanks to the following contributors
       <td align="center" valign="top" width="14.28%"><a href="http://www.vanderplas.com/"><img src="https://avatars.githubusercontent.com/u/781659?v=4?s=100" width="100px;" alt="Jake Vanderplas"/><br /><sub><b>Jake Vanderplas</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/pulls?q=is%3Apr+reviewed-by%3Ajakevdp" title="Reviewed Pull Requests">👀</a></td>
       <td align="center" valign="top" width="14.28%"><a href="https://github.com/KotlinIsland"><img src="https://avatars.githubusercontent.com/u/65446343?v=4?s=100" width="100px;" alt="kotlinisland"/><br /><sub><b>kotlinisland</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/pulls?q=is%3Apr+reviewed-by%3Akotlinisland" title="Reviewed Pull Requests">👀</a></td>
       <td align="center" valign="top" width="14.28%"><a href="https://github.com/DetachHead"><img src="https://avatars.githubusercontent.com/u/57028336?v=4?s=100" width="100px;" alt="detachhead"/><br /><sub><b>detachhead</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/pulls?q=is%3Apr+reviewed-by%3Adetachhead" title="Reviewed Pull Requests">👀</a></td>
-      <td align="center" valign="top" width="14.28%"><a href="https://github.com/NeilGirdhar"><img src="https://avatars.githubusercontent.com/u/730137?v=4?s=100" width="100px;" alt="Neil Girdhar"/><br /><sub><b>Neil Girdhar</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/commits?author=NeilGirdhar" title="Code">💻</a> <a href="https://github.com/data-apis/array-api-extra/commits?author=NeilGirdhar" title="Tests">⚠️</a></td>
+      <td align="center" valign="top" width="14.28%"><a href="https://github.com/NeilGirdhar"><img src="https://avatars.githubusercontent.com/u/730137?v=4?s=100" width="100px;" alt="Neil Girdhar"/><br /><sub><b>Neil Girdhar</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/commits?author=NeilGirdhar" title="Code">💻</a> <a href="https://github.com/data-apis/array-api-extra/commits?author=NeilGirdhar" title="Tests">⚠️</a> <a href="https://github.com/data-apis/array-api-extra/pulls?q=is%3Apr+reviewed-by%3ANeilGirdhar" title="Reviewed Pull Requests">👀</a></td>
     </tr>
     <tr>
       <td align="center" valign="top" width="14.28%"><a href="https://github.com/lithomas1"><img src="https://avatars.githubusercontent.com/u/47963215?v=4?s=100" width="100px;" alt="Thomas Li"/><br /><sub><b>Thomas Li</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/issues?q=author%3Alithomas1" title="Bug reports">🐛</a> <a href="#tool-lithomas1" title="Tools">🔧</a></td>
@@ -46,6 +46,8 @@ This project exists thanks to the following contributors
     </tr>
     <tr>
       <td align="center" valign="top" width="14.28%"><a href="http://steppi.github.io"><img src="https://avatars.githubusercontent.com/u/1953382?v=4?s=100" width="100px;" alt="Albert Steppi"/><br /><sub><b>Albert Steppi</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/commits?author=steppi" title="Code">💻</a> <a href="#ideas-steppi" title="Ideas, Planning, & Feedback">🤔</a> <a href="https://github.com/data-apis/array-api-extra/commits?author=steppi" title="Tests">⚠️</a></td>
+      <td align="center" valign="top" width="14.28%"><a href="https://github.com/prady0t"><img src="https://avatars.githubusercontent.com/u/99216956?v=4?s=100" width="100px;" alt="Pradyot Ranjan"/><br /><sub><b>Pradyot Ranjan</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/commits?author=prady0t" title="Code">💻</a> <a href="https://github.com/data-apis/array-api-extra/commits?author=prady0t" title="Documentation">📖</a> <a href="#ideas-prady0t" title="Ideas, Planning, & Feedback">🤔</a> <a href="https://github.com/data-apis/array-api-extra/commits?author=prady0t" title="Tests">⚠️</a></td>
+      <td align="center" valign="top" width="14.28%"><a href="https://github.com/lundybernard"><img src="https://avatars.githubusercontent.com/u/17297695?v=4?s=100" width="100px;" alt="Lundy Bernard"/><br /><sub><b>Lundy Bernard</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/pulls?q=is%3Apr+reviewed-by%3Alundybernard" title="Reviewed Pull Requests">👀</a></td>
     </tr>
   </tbody>
 </table>

{array_api_extra-0.10.1 → array_api_extra-0.10.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: array-api-extra
-Version: 0.10.1
+Version: 0.10.3
 Summary: Extra array functions built on top of the array API standard.
 Author-Email: Lucas Colley <lucas.colley8@gmail.com>
 License-Expression: MIT
@@ -20,7 +20,7 @@ Project-URL: Homepage, https://github.com/data-apis/array-api-extra
 Project-URL: Bug Tracker, https://github.com/data-apis/array-api-extra/issues
 Project-URL: Changelog, https://github.com/data-apis/array-api-extra/releases
 Requires-Python: >=3.11
-Requires-Dist: array-api-compat<2,>=1.13.0
+Requires-Dist: array-api-compat<2,>=1.14.0
 Description-Content-Type: text/markdown
 
 # array-api-extra
@@ -61,8 +61,10 @@ Used by:
 
 - [SciPy](https://github.com/scipy/scipy) — Fundamental algorithms for
   scientific computing.
-- [scikit-learn](https://github.com/scikit-learn/scikit-learn) — Machine
+- [scikit-learn](https://github.com/scikit-learn/scikit-learn) — Machine
   Learning in Python.
+- [SysIdentPy](https://github.com/wilsonrljr/sysidentpy) — System Identification Using NARMAX Models.
+- [GLASS](https://github.com/glass-dev/glass) — Generator for Large Scale Structure.
 - *your library? Let us know!*
 
 ## Installation
@@ -113,7 +115,7 @@ This project exists thanks to the following contributors
       <td align="center" valign="top" width="14.28%"><a href="http://www.vanderplas.com/"><img src="https://avatars.githubusercontent.com/u/781659?v=4?s=100" width="100px;" alt="Jake Vanderplas"/><br /><sub><b>Jake Vanderplas</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/pulls?q=is%3Apr+reviewed-by%3Ajakevdp" title="Reviewed Pull Requests">👀</a></td>
       <td align="center" valign="top" width="14.28%"><a href="https://github.com/KotlinIsland"><img src="https://avatars.githubusercontent.com/u/65446343?v=4?s=100" width="100px;" alt="kotlinisland"/><br /><sub><b>kotlinisland</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/pulls?q=is%3Apr+reviewed-by%3Akotlinisland" title="Reviewed Pull Requests">👀</a></td>
       <td align="center" valign="top" width="14.28%"><a href="https://github.com/DetachHead"><img src="https://avatars.githubusercontent.com/u/57028336?v=4?s=100" width="100px;" alt="detachhead"/><br /><sub><b>detachhead</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/pulls?q=is%3Apr+reviewed-by%3Adetachhead" title="Reviewed Pull Requests">👀</a></td>
-      <td align="center" valign="top" width="14.28%"><a href="https://github.com/NeilGirdhar"><img src="https://avatars.githubusercontent.com/u/730137?v=4?s=100" width="100px;" alt="Neil Girdhar"/><br /><sub><b>Neil Girdhar</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/commits?author=NeilGirdhar" title="Code">💻</a> <a href="https://github.com/data-apis/array-api-extra/commits?author=NeilGirdhar" title="Tests">⚠️</a></td>
+      <td align="center" valign="top" width="14.28%"><a href="https://github.com/NeilGirdhar"><img src="https://avatars.githubusercontent.com/u/730137?v=4?s=100" width="100px;" alt="Neil Girdhar"/><br /><sub><b>Neil Girdhar</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/commits?author=NeilGirdhar" title="Code">💻</a> <a href="https://github.com/data-apis/array-api-extra/commits?author=NeilGirdhar" title="Tests">⚠️</a> <a href="https://github.com/data-apis/array-api-extra/pulls?q=is%3Apr+reviewed-by%3ANeilGirdhar" title="Reviewed Pull Requests">👀</a></td>
     </tr>
     <tr>
       <td align="center" valign="top" width="14.28%"><a href="https://github.com/lithomas1"><img src="https://avatars.githubusercontent.com/u/47963215?v=4?s=100" width="100px;" alt="Thomas Li"/><br /><sub><b>Thomas Li</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/issues?q=author%3Alithomas1" title="Bug reports">🐛</a> <a href="#tool-lithomas1" title="Tools">🔧</a></td>
@@ -126,6 +128,8 @@ This project exists thanks to the following contributors
     </tr>
     <tr>
       <td align="center" valign="top" width="14.28%"><a href="http://steppi.github.io"><img src="https://avatars.githubusercontent.com/u/1953382?v=4?s=100" width="100px;" alt="Albert Steppi"/><br /><sub><b>Albert Steppi</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/commits?author=steppi" title="Code">💻</a> <a href="#ideas-steppi" title="Ideas, Planning, & Feedback">🤔</a> <a href="https://github.com/data-apis/array-api-extra/commits?author=steppi" title="Tests">⚠️</a></td>
+      <td align="center" valign="top" width="14.28%"><a href="https://github.com/prady0t"><img src="https://avatars.githubusercontent.com/u/99216956?v=4?s=100" width="100px;" alt="Pradyot Ranjan"/><br /><sub><b>Pradyot Ranjan</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/commits?author=prady0t" title="Code">💻</a> <a href="https://github.com/data-apis/array-api-extra/commits?author=prady0t" title="Documentation">📖</a> <a href="#ideas-prady0t" title="Ideas, Planning, & Feedback">🤔</a> <a href="https://github.com/data-apis/array-api-extra/commits?author=prady0t" title="Tests">⚠️</a></td>
+      <td align="center" valign="top" width="14.28%"><a href="https://github.com/lundybernard"><img src="https://avatars.githubusercontent.com/u/17297695?v=4?s=100" width="100px;" alt="Lundy Bernard"/><br /><sub><b>Lundy Bernard</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/pulls?q=is%3Apr+reviewed-by%3Alundybernard" title="Reviewed Pull Requests">👀</a></td>
     </tr>
   </tbody>
 </table>

{array_api_extra-0.10.1 → array_api_extra-0.10.3}/README.md

@@ -36,8 +36,10 @@ Used by:
 
 - [SciPy](https://github.com/scipy/scipy) — Fundamental algorithms for
   scientific computing.
-- [scikit-learn](https://github.com/scikit-learn/scikit-learn) — Machine
+- [scikit-learn](https://github.com/scikit-learn/scikit-learn) — Machine
   Learning in Python.
+- [SysIdentPy](https://github.com/wilsonrljr/sysidentpy) — System Identification Using NARMAX Models.
+- [GLASS](https://github.com/glass-dev/glass) — Generator for Large Scale Structure.
 - *your library? Let us know!*
 
 ## Installation
@@ -88,7 +90,7 @@ This project exists thanks to the following contributors
       <td align="center" valign="top" width="14.28%"><a href="http://www.vanderplas.com/"><img src="https://avatars.githubusercontent.com/u/781659?v=4?s=100" width="100px;" alt="Jake Vanderplas"/><br /><sub><b>Jake Vanderplas</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/pulls?q=is%3Apr+reviewed-by%3Ajakevdp" title="Reviewed Pull Requests">👀</a></td>
       <td align="center" valign="top" width="14.28%"><a href="https://github.com/KotlinIsland"><img src="https://avatars.githubusercontent.com/u/65446343?v=4?s=100" width="100px;" alt="kotlinisland"/><br /><sub><b>kotlinisland</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/pulls?q=is%3Apr+reviewed-by%3Akotlinisland" title="Reviewed Pull Requests">👀</a></td>
       <td align="center" valign="top" width="14.28%"><a href="https://github.com/DetachHead"><img src="https://avatars.githubusercontent.com/u/57028336?v=4?s=100" width="100px;" alt="detachhead"/><br /><sub><b>detachhead</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/pulls?q=is%3Apr+reviewed-by%3Adetachhead" title="Reviewed Pull Requests">👀</a></td>
-      <td align="center" valign="top" width="14.28%"><a href="https://github.com/NeilGirdhar"><img src="https://avatars.githubusercontent.com/u/730137?v=4?s=100" width="100px;" alt="Neil Girdhar"/><br /><sub><b>Neil Girdhar</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/commits?author=NeilGirdhar" title="Code">💻</a> <a href="https://github.com/data-apis/array-api-extra/commits?author=NeilGirdhar" title="Tests">⚠️</a></td>
+      <td align="center" valign="top" width="14.28%"><a href="https://github.com/NeilGirdhar"><img src="https://avatars.githubusercontent.com/u/730137?v=4?s=100" width="100px;" alt="Neil Girdhar"/><br /><sub><b>Neil Girdhar</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/commits?author=NeilGirdhar" title="Code">💻</a> <a href="https://github.com/data-apis/array-api-extra/commits?author=NeilGirdhar" title="Tests">⚠️</a> <a href="https://github.com/data-apis/array-api-extra/pulls?q=is%3Apr+reviewed-by%3ANeilGirdhar" title="Reviewed Pull Requests">👀</a></td>
     </tr>
     <tr>
       <td align="center" valign="top" width="14.28%"><a href="https://github.com/lithomas1"><img src="https://avatars.githubusercontent.com/u/47963215?v=4?s=100" width="100px;" alt="Thomas Li"/><br /><sub><b>Thomas Li</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/issues?q=author%3Alithomas1" title="Bug reports">🐛</a> <a href="#tool-lithomas1" title="Tools">🔧</a></td>
@@ -101,6 +103,8 @@ This project exists thanks to the following contributors
     </tr>
     <tr>
       <td align="center" valign="top" width="14.28%"><a href="http://steppi.github.io"><img src="https://avatars.githubusercontent.com/u/1953382?v=4?s=100" width="100px;" alt="Albert Steppi"/><br /><sub><b>Albert Steppi</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/commits?author=steppi" title="Code">💻</a> <a href="#ideas-steppi" title="Ideas, Planning, & Feedback">🤔</a> <a href="https://github.com/data-apis/array-api-extra/commits?author=steppi" title="Tests">⚠️</a></td>
+      <td align="center" valign="top" width="14.28%"><a href="https://github.com/prady0t"><img src="https://avatars.githubusercontent.com/u/99216956?v=4?s=100" width="100px;" alt="Pradyot Ranjan"/><br /><sub><b>Pradyot Ranjan</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/commits?author=prady0t" title="Code">💻</a> <a href="https://github.com/data-apis/array-api-extra/commits?author=prady0t" title="Documentation">📖</a> <a href="#ideas-prady0t" title="Ideas, Planning, & Feedback">🤔</a> <a href="https://github.com/data-apis/array-api-extra/commits?author=prady0t" title="Tests">⚠️</a></td>
+      <td align="center" valign="top" width="14.28%"><a href="https://github.com/lundybernard"><img src="https://avatars.githubusercontent.com/u/17297695?v=4?s=100" width="100px;" alt="Lundy Bernard"/><br /><sub><b>Lundy Bernard</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/pulls?q=is%3Apr+reviewed-by%3Alundybernard" title="Reviewed Pull Requests">👀</a></td>
     </tr>
   </tbody>
 </table>

array_api_extra-0.10.3/docs/_templates/autosummary/class.rst

@@ -0,0 +1,5 @@
+{{ name | escape | underline }}
+
+.. currentmodule:: {{ module }}
+
+.. autoclass:: {{ fullname }}

array_api_extra-0.10.3/docs/_templates/autosummary/function.rst

@@ -0,0 +1,5 @@
+{{ name | escape | underline }}
+
+.. currentmodule:: {{ module }}
+
+.. autofunction:: {{ fullname }}

array_api_extra-0.10.3/docs/_templates/testing-function.rst

@@ -0,0 +1,5 @@
+{{ fullname.split('.')[-2:] | join('.') | escape | underline }}
+
+.. currentmodule:: {{ module }}
+
+.. autofunction:: {{ fullname }}

array_api_extra-0.10.1/docs/api-reference.md → array_api_extra-0.10.3/docs/api-assorted.md

@@ -1,4 +1,4 @@
-# API Reference
+# Assorted functions
 
 ```{eval-rst}
 .. currentmodule:: array_api_extra
@@ -6,6 +6,7 @@
     :nosignatures:
     :toctree: generated
 
+    angle
     apply_where
     argpartition
     at

array_api_extra-0.10.3/docs/api-lazy.md

@@ -0,0 +1,26 @@
+# Tools for lazy backends
+
+Additional functions with special behaviour for lazy backends,
+e.g. Dask or JAX:
+
+```{eval-rst}
+.. currentmodule:: array_api_extra
+.. autosummary::
+    :nosignatures:
+    :toctree: generated
+
+    lazy_apply
+```
+
+Testing utilities:
+
+```{eval-rst}
+.. currentmodule:: array_api_extra
+.. autosummary::
+    :nosignatures:
+    :template: testing-function.rst
+    :toctree: generated
+
+    testing.lazy_xp_function
+    testing.patch_lazy_xp_functions
+```

array_api_extra-0.10.3/docs/api-testing.md

@@ -0,0 +1,19 @@
+# Testing utilities (`xpx.testing`)
+
+These testing utilities are intended to help transition from using
+{py:mod}`numpy.testing`.
+
+```{eval-rst}
+.. currentmodule:: array_api_extra.testing
+.. autosummary::
+    :nosignatures:
+    :template: testing-function.rst
+    :toctree: generated
+
+    assert_close
+    assert_close_nulp
+    assert_equal
+    assert_less
+```
+
+See also utilities specific to testing lazy backends in {doc}`api-lazy`.

{array_api_extra-0.10.1 → array_api_extra-0.10.3}/docs/conf.py

@@ -66,4 +66,6 @@ nitpick_ignore = [
     ("py:class", "_io.BytesIO"),
 ]
 
+templates_path = ["_templates"]
+
 always_document_param_types = True

{array_api_extra-0.10.1 → array_api_extra-0.10.3}/docs/index.md

@@ -4,7 +4,8 @@
 :maxdepth: 2
 :hidden:
 self
-api-reference.md
+api-assorted.md
+api-testing.md
 api-lazy.md
 contributing.md
 contributors.md
@@ -27,6 +28,8 @@ It is currently used by:
   scientific computing.
 - [scikit-learn](https://github.com/scikit-learn/scikit-learn) — Machine
   Learning in Python.
+- [SysIdentPy](https://github.com/wilsonrljr/sysidentpy) — System Identification Using NARMAX Models.
+- [GLASS](https://github.com/glass-dev/glass) — Generator for Large Scale Structure.
 - *your library? Let us know!*
 
 (installation)=
@@ -114,7 +117,7 @@ def array_namespace(*xs, **kwargs):
 See [an example of this in SciPy][scipy-vendor-example].
 ```
 
-[scipy-vendor-example]: https://github.com/scipy/scipy/blob/main/scipy/_lib/_array_api_compat_vendor.py
+[scipy-vendor-example]: https://github.com/scipy/scipy/blob/main/scipy/_external/_array_api_compat_vendor.py
 
 ## Versioning
 

{array_api_extra-0.10.1 → array_api_extra-0.10.3}/lefthook.yml

@@ -39,11 +39,16 @@ pre-commit:
     - name: mypy
       glob: "*.{py,pyi}"
       run: pixi {run} mypy
+    - name: pyrefly
+      glob: "*.{py, pyi}"
+      run: pixi {run} pyrefly
     - name: typos
       stage_fixed: true
       run: pixi {run} typos
     - name: actionlint
       run: pixi {run} actionlint
+    - name: zizmor
+      run: pixi {run} zizmor
     - name: blacken-docs
       glob: "*.md"
       stage_fixed: true

{array_api_extra-0.10.1 → array_api_extra-0.10.3}/meson.build

@@ -1,6 +1,6 @@
 project(
   'array-api-extra',
-  version: '0.10.1'
+  version: '0.10.3'
 )
 
 py = import('python').find_installation()

array_api_extra-0.10.3/pixi.toml

@@ -0,0 +1,219 @@
+[workspace]
+channels = ["https://prefix.dev/conda-forge"]
+platforms = ["linux-64", "osx-64", "osx-arm64", "win-64"]
+preview = ["pixi-build"]
+requires-pixi = ">=0.68.0"
+
+### array-api-extra package definition ###
+
+[package.build.backend]
+name = "pixi-build-python"
+version = "*"
+
+[package.host-dependencies]
+meson-python = "*"
+uv = "*" # interfaces with meson-python instead of pip
+
+[package.run-dependencies]
+array-api-compat = "*"
+
+### workspace environments ###
+
+[environments]
+default = { features = ["py314"], solve-group = "py314" }
+lint = { features = ["py314", "lint"], solve-group = "py314" }
+docs = { features = ["py314", "docs"], solve-group = "py314" }
+tests = { features = ["py314", "tests"], solve-group = "py314" }
+tests-py314 = { features = ["py314", "tests"], solve-group = "py314" } # alias of tests
+tests-run-deps = { features = ["py314", "tests-run-deps"], solve-group = "py314" }
+
+# Some backends may pin numpy; use separate solve-group
+dev = { features = ["py314", "lint", "tests", "docs", "dev", "backends"], solve-group = "backends" }
+tests-backends = { features = ["py314", "tests", "backends"], solve-group = "backends" }
+tests-backends-py311 = { features = ["py311", "tests", "backends"] }
+
+# CUDA not available on free github actions and on some developers' PCs
+dev-cuda = { features = ["py314", "lint", "tests", "docs", "dev", "backends", "cuda-backends"], solve-group = "cuda" }
+tests-cuda = { features = ["py314", "tests", "backends", "cuda-backends"], solve-group = "cuda" }
+tests-cuda-py311 = { features = ["py311", "tests", "backends", "cuda-backends"] }
+
+# Ungrouped environments
+tests-numpy1 = ["py311", "tests", "numpy1"]
+tests-py311 = ["py311", "tests"]
+tests-nogil = ["nogil", "tests"]
+
+### default feature definition ###
+
+[dev]
+# this pulls in array-api-extra's host and run dependencies
+array-api-extra.path = "."
+
+[dependencies]
+array-api-extra.path = "."
+
+### non-default feature definitions ###
+
+[feature.lint.dependencies]
+typing-extensions = ">=4.15.0"
+pylint = ">=4.0.5"
+mypy = ">=1.20.2"
+basedpyright = ">=1.39.5"
+numpydoc = ">=1.10.0,<2"
+# import dependencies for mypy:
+array-api-strict = ">=2.5,<2.6"
+numpy = ">=2.1.3"
+hypothesis = ">=6.152.2"
+dask-core = ">=2026.3.0" # No distributed, tornado, etc.
+dprint = ">=0.50.0,<0.51"
+lefthook = ">=2.1.8,<3"
+ruff = ">=0.15.13,<0.16"
+typos = ">=1.46.2,<2"
+actionlint = ">=1.7.12,<2"
+blacken-docs = ">=1.20.0,<2"
+pytest = ">=9.0.3,<10"
+validate-pyproject = ">=0.25,<0.26"
+pyrefly = ">=1.0.0,<1.1"
+zizmor = ">=1.25.2,<1.26"
+# NOTE: don't add cupy, jax, pytorch, or sparse here,
+# as they slow down mypy and are not portable across target OSs
+
+[feature.lint.tasks]
+lefthook = { cmd = "lefthook", description = "Run lefthook", default-environment = "lint" }
+hooks = { cmd = "lefthook install", description = "Install pre-commit hooks", default-environment = "lint" }
+pre-commit = { cmd = "lefthook run pre-commit", description = "Run pre-commit checks", default-environment = "lint" }
+pylint = { cmd = "pylint array_api_extra", cwd = "src", description = "Lint with pylint", default-environment = "lint" }
+mypy = { cmd = "mypy", description = "Type check with mypy", default-environment = "lint" }
+pyrefly = { cmd = "pyrefly check", description = "Type check with pyrefly", default-environment = "lint" }
+pyright = { cmd = "basedpyright", description = "Type check with basedpyright", default-environment = "lint" }
+ruff-check = { cmd = "ruff check --fix", description = "Lint with ruff", default-environment = "lint" }
+ruff-format = { cmd = "ruff format", description = "Format with ruff", default-environment = "lint" }
+dprint = { cmd = "dprint fmt", description = "Format with dprint", default-environment = "lint" }
+typos = { cmd = "typos --write-changes --force-exclude", description = "Fix typos", default-environment = "lint" }
+actionlint = { cmd = "actionlint", description = "Lint actions with actionlint", default-environment = "lint" }
+zizmor = { cmd = "zizmor .github -p", description = "GHA static analysis with zizmor", default-environment = "lint" }
+blacken-docs = { cmd = "blacken-docs", description = "Format Python markdown blocks with Black", default-environment = "lint" }
+validate-pyproject = { cmd = "validate-pyproject pyproject.toml", description = "Validate pyproject.toml", default-environment = "lint" }
+numpydoc = { cmd = "numpydoc lint", description = "Validate docstrings with numpydoc", default-environment = "lint" }
+lint = { cmd = "lefthook run pre-commit --all-files --force", description = "Run all linters", default-environment = "lint" }
+
+[feature.tests.dependencies]
+pytest = ">=9.0.3"
+pytest-cov = ">=7.1.0"
+hypothesis = ">=6.152.2"
+array-api-strict = ">=2.5,<2.6"
+numpy = ">=1.22.0"
+scipy = ">=1.15.2,<2"
+
+[feature.tests.tasks]
+tests = { cmd = "pytest -v", description = "Run tests", default-environment = "tests" }
+tests-cov = { cmd = "pytest -v -ra --cov --cov-report=xml --cov-report=term --durations=20", description = "Run tests with coverage", default-environment = "tests" }
+
+clean-vendor-compat = { cmd = "rm -rf vendor_tests/array_api_compat", description = "Delete the existing vendored version of array-api-compat", default-environment = "tests" }
+clean-vendor-extra = { cmd = "rm -rf vendor_tests/array_api_extra", description = "Delete the existing vendored version of array-api-extra", default-environment = "tests" }
+copy-vendor-compat = { cmd = "cp -r $(python -c 'import site; print(site.getsitepackages()[0])')/array_api_compat vendor_tests/", depends-on = ["clean-vendor-compat"], description = "Vendor a clean copy of array-api-compat", default-environment = "tests" }
+copy-vendor-extra = { cmd = "cp -r src/array_api_extra vendor_tests/", depends-on = ["clean-vendor-extra"], description = "Vendor a clean copy of array-api-extra", default-environment = "tests" }
+tests-vendor = { cmd = "pytest -v vendor_tests", depends-on = ["copy-vendor-compat", "copy-vendor-extra"], description = "Check that array-api-extra and array-api-compat can be vendored together", default-environment = "tests" }
+
+tests-ci = { depends-on = ["tests-cov", "tests-vendor"], description = "Run tests with coverage and vendor tests" }
+coverage = { cmd = "coverage html", depends-on = ["tests-cov"], description = "Generate test coverage html report", default-environment = "tests" }
+open-coverage = { cmd = "open htmlcov/index.html", depends-on = ["coverage"], description = "Open test coverage report", default-environment = "tests" }
+
+# tests with only run dependencies
+[feature.tests-run-deps.dependencies]
+pytest = ">=9.0.3"
+pytest-cov = ">=7.1.0"
+
+[feature.tests-run-deps.tasks.tests-run-deps]
+cmd = "pytest -v run_deps_tests"
+description = "Run run-dependency tests"
+default-environment = "tests-run-deps"
+
+[feature.tests-run-deps.tasks.tests-run-deps-cov]
+cmd = "pytest -v -ra --cov --cov-report=xml --cov-report=term --durations=20 run_deps_tests"
+description = "Run run-dependency tests with coverage"
+default-environment = "tests-run-deps"
+
+[feature.docs.dependencies]
+sphinx = ">=7.4.7"
+furo = ">=2025.12.19"
+myst-parser = ">=5.1.0"
+sphinx-copybutton = ">=0.5.2"
+sphinx-autodoc-typehints = ">=1.25.3"
+# Needed to import parsed modules with autodoc
+dask-core = ">=2026.3.0" # No distributed, tornado, etc.
+pytest = ">=9.0.3"
+typing-extensions = ">=4.15.0"
+numpy = ">=2.1.3"
+
+[feature.docs.tasks]
+docs = { cmd = "sphinx-build -E -W . build/", cwd = "docs", description = "Build docs", default-environment = "docs" }
+open-docs = { cmd = "open build/index.html", cwd = "docs", depends-on = ["docs"], description = "Open the generated docs", default-environment = "docs" }
+
+[feature.dev.dependencies]
+ipython = ">=9.13.0"
+
+[feature.dev.tasks]
+ipython = { cmd = "ipython", description = "Launch ipython", default-environment = "dev" }
+
+[feature.py311.dependencies]
+python = "~=3.11.0"
+
+[feature.py314.dependencies]
+python = "~=3.14.0"
+
+[feature.numpy1.dependencies]
+# Oldest NumPy version supported by scikit-learn.
+# Note that this is older than what SPEC0 recommends.
+numpy = "=1.24.1"
+
+# Backends that can run on CPU-only hosts
+# Note: JAX and PyTorch will install CPU variants.
+[feature.backends.dependencies]
+pytorch = ">=2.10.0"
+dask-core = ">=2026.3.0" # No distributed, tornado, etc.
+sparse = ">=0.18.0"
+
+[feature.backends.target.linux-64.dependencies]
+jax = ">=0.9.2"
+
+[feature.backends.target.osx-64.dependencies]
+jax = ">=0.9.2"
+
+[feature.backends.target.osx-arm64.dependencies]
+jax = ">=0.9.2"
+
+[feature.backends.target.win-64.dependencies]
+# jax = "*"  # unavailable
+
+# Backends that require a GPU host and a CUDA driver.
+# Note that JAX and PyTorch automatically prefer CUDA variants
+# thanks to the `system-requirements` below, *if available*.
+# We request them explicitly below to ensure that we don't
+# quietly revert to CPU-only in the future, e.g. when CUDA 13
+# is released and CUDA 12 builds are dropped upstream.
+[feature.cuda-backends]
+system-requirements = { cuda = "12" }
+
+[feature.cuda-backends.target.linux.dependencies]
+cupy = ">=14.0.1"
+jaxlib = { version = ">=0.9.2", build = "cuda12*" }
+pytorch = { version = ">=2.10.0", build = "cuda12*" }
+
+[feature.cuda-backends.target.osx.dependencies]
+# cupy = "*"  # unavailable
+# jaxlib = { version = "*", build = "cuda12*" }  # unavailable
+# pytorch = { version = "*", build = "cuda12*" }  # unavailable
+
+[feature.cuda-backends.target.win.dependencies]
+cupy = ">=14.0.1"
+# jaxlib = { version = "*", build = "cuda12*" }  # unavailable
+pytorch = { version = ">=2.10.0", build = "cuda12*" }
+
+[feature.nogil.dependencies]
+python-freethreading = "~=3.13.0"
+pytest-run-parallel = ">=0.8.2"
+numpy = ">=2.3.5"
+# pytorch = "*"  # Not available on Python 3.13t yet
+dask-core = ">=2026.3.0" # No distributed, tornado, etc.
+# sparse = "*"  # numba not available on Python 3.13t yet
+# jax = "*"  # ml_dtypes not available on Python 3.13t yet