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

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

@@ -308,6 +308,17 @@
         "ideas",
         "test"
       ]
+    },
+    {
+      "login": "steppi",
+      "name": "Albert Steppi",
+      "avatar_url": "https://avatars.githubusercontent.com/u/1953382?v=4",
+      "profile": "http://steppi.github.io",
+      "contributions": [
+        "code",
+        "ideas",
+        "test"
+      ]
     }
   ]
 }

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

@@ -44,6 +44,9 @@ This project exists thanks to the following contributors
       <td align="center" valign="top" width="14.28%"><a href="https://github.com/adriagarp"><img src="https://avatars.githubusercontent.com/u/96059447?v=4?s=100" width="100px;" alt="Adrián García Pitarch"/><br /><sub><b>Adrián García Pitarch</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/commits?author=adriagarp" title="Code">💻</a> <a href="https://github.com/data-apis/array-api-extra/commits?author=adriagarp" title="Tests">⚠️</a></td>
       <td align="center" valign="top" width="14.28%"><a href="https://github.com/cakedev0"><img src="https://avatars.githubusercontent.com/u/25986961?v=4?s=100" width="100px;" alt="Arthur Lacote"/><br /><sub><b>Arthur Lacote</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/commits?author=cakedev0" title="Code">💻</a> <a href="https://github.com/data-apis/array-api-extra/commits?author=cakedev0" title="Documentation">📖</a> <a href="#ideas-cakedev0" title="Ideas, Planning, & Feedback">🤔</a> <a href="https://github.com/data-apis/array-api-extra/commits?author=cakedev0" title="Tests">⚠️</a></td>
     </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>
+    </tr>
   </tbody>
 </table>
 

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

@@ -1,46 +1,26 @@
 Metadata-Version: 2.4
 Name: array-api-extra
-Version: 0.9.2
+Version: 0.10.1
 Summary: Extra array functions built on top of the array API standard.
-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
-Author-email: Lucas Colley <lucas.colley8@gmail.com>
-License: MIT License
-        
-        Copyright (c) 2024 Consortium for Python Data API Standards
-        
-        Permission is hereby granted, free of charge, to any person obtaining a copy
-        of this software and associated documentation files (the "Software"), to deal
-        in the Software without restriction, including without limitation the rights
-        to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-        copies of the Software, and to permit persons to whom the Software is
-        furnished to do so, subject to the following conditions:
-        
-        The above copyright notice and this permission notice shall be included in all
-        copies or substantial portions of the Software.
-        
-        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-        AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-        LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-        OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-        SOFTWARE.
+Author-Email: Lucas Colley <lucas.colley8@gmail.com>
+License-Expression: MIT
 License-File: LICENSE
 Classifier: Intended Audience :: Developers
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3 :: Only
-Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Classifier: Programming Language :: Python :: Free Threading :: 3 - Stable
 Classifier: Typing :: Typed
-Requires-Python: >=3.10
-Requires-Dist: array-api-compat<2,>=1.12.0
+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
 Description-Content-Type: text/markdown
 
 # array-api-extra
@@ -144,6 +124,9 @@ This project exists thanks to the following contributors
       <td align="center" valign="top" width="14.28%"><a href="https://github.com/adriagarp"><img src="https://avatars.githubusercontent.com/u/96059447?v=4?s=100" width="100px;" alt="Adrián García Pitarch"/><br /><sub><b>Adrián García Pitarch</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/commits?author=adriagarp" title="Code">💻</a> <a href="https://github.com/data-apis/array-api-extra/commits?author=adriagarp" title="Tests">⚠️</a></td>
       <td align="center" valign="top" width="14.28%"><a href="https://github.com/cakedev0"><img src="https://avatars.githubusercontent.com/u/25986961?v=4?s=100" width="100px;" alt="Arthur Lacote"/><br /><sub><b>Arthur Lacote</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/commits?author=cakedev0" title="Code">💻</a> <a href="https://github.com/data-apis/array-api-extra/commits?author=cakedev0" title="Documentation">📖</a> <a href="#ideas-cakedev0" title="Ideas, Planning, & Feedback">🤔</a> <a href="https://github.com/data-apis/array-api-extra/commits?author=cakedev0" title="Tests">⚠️</a></td>
     </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>
+    </tr>
   </tbody>
 </table>
 

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

@@ -99,6 +99,9 @@ This project exists thanks to the following contributors
       <td align="center" valign="top" width="14.28%"><a href="https://github.com/adriagarp"><img src="https://avatars.githubusercontent.com/u/96059447?v=4?s=100" width="100px;" alt="Adrián García Pitarch"/><br /><sub><b>Adrián García Pitarch</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/commits?author=adriagarp" title="Code">💻</a> <a href="https://github.com/data-apis/array-api-extra/commits?author=adriagarp" title="Tests">⚠️</a></td>
       <td align="center" valign="top" width="14.28%"><a href="https://github.com/cakedev0"><img src="https://avatars.githubusercontent.com/u/25986961?v=4?s=100" width="100px;" alt="Arthur Lacote"/><br /><sub><b>Arthur Lacote</b></sub></a><br /><a href="https://github.com/data-apis/array-api-extra/commits?author=cakedev0" title="Code">💻</a> <a href="https://github.com/data-apis/array-api-extra/commits?author=cakedev0" title="Documentation">📖</a> <a href="#ideas-cakedev0" title="Ideas, Planning, & Feedback">🤔</a> <a href="https://github.com/data-apis/array-api-extra/commits?author=cakedev0" title="Tests">⚠️</a></td>
     </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>
+    </tr>
   </tbody>
 </table>
 

{array_api_extra-0.9.2 → array_api_extra-0.10.1}/docs/contributing.md

@@ -20,6 +20,8 @@ Thanks to [all contributors](contributors.md) so far!
 - Import your function to `src/array_api_extra/__init__.py` and add it to
   `__all__` there.
 - Add a test class for your function in `tests/test_funcs.py`.
+  - Ensure that `lazy_xp_function` is called on the function if lazy backends
+    are supposed to be tested.
 - Add your function to `docs/api-reference.md`.
 - [Make a PR!](https://github.com/data-apis/array-api-extra/pulls)
 

array_api_extra-0.10.1/meson.build

@@ -0,0 +1,39 @@
+project(
+  'array-api-extra',
+  version: '0.10.1'
+)
+
+py = import('python').find_installation()
+
+# NOTE: downstream projects may rely on this variable name when vendoring,
+# do not change it without consulting with downstream projects
+sources = {
+  'array_api_extra': files(
+    'src/array_api_extra/__init__.py',
+    'src/array_api_extra/_delegation.py',
+    'src/array_api_extra/testing.py',
+  ),
+  'array_api_extra/_lib': files(
+    'src/array_api_extra/_lib/__init__.py',
+    'src/array_api_extra/_lib/_at.py',
+    'src/array_api_extra/_lib/_backends.py',
+    'src/array_api_extra/_lib/_funcs.py',
+    'src/array_api_extra/_lib/_lazy.py',
+    'src/array_api_extra/_lib/_testing.py',
+  ),
+  'array_api_extra/_lib/_utils': files(
+    'src/array_api_extra/_lib/_utils/__init__.py',
+    'src/array_api_extra/_lib/_utils/_compat.py',
+    'src/array_api_extra/_lib/_utils/_compat.pyi',
+    'src/array_api_extra/_lib/_utils/_helpers.py',
+    'src/array_api_extra/_lib/_utils/_typing.py',
+    'src/array_api_extra/_lib/_utils/_typing.pyi',
+  ),
+}
+
+foreach subdir, files : sources
+  py.install_sources(files, subdir: subdir)
+endforeach
+
+subdir('tests')
+subdir('vendor_tests')

{array_api_extra-0.9.2 → array_api_extra-0.10.1}/pyproject.toml

@@ -1,77 +1,113 @@
 [build-system]
-requires = ["hatchling"]
-build-backend = "hatchling.build"
+requires = ["meson-python"]
+build-backend = "mesonpy"
 
 [project]
 name = "array-api-extra"
+version = "0.10.1"
 authors = [
   { name = "Lucas Colley", email = "lucas.colley8@gmail.com" },
   # { name = "Open Source Contributors" },  # https://github.com/pypi/warehouse/issues/14813
 ]
 description = "Extra array functions built on top of the array API standard."
 readme = "README.md"
-license.file = "LICENSE"
-requires-python = ">=3.10"
+license = "MIT"
+license-files = ["LICENSE"]
+requires-python = ">=3.11"
 classifiers = [
   "Intended Audience :: Developers",
   "Operating System :: OS Independent",
   "Programming Language :: Python",
   "Programming Language :: Python :: 3",
   "Programming Language :: Python :: 3 :: Only",
-  "Programming Language :: Python :: 3.10",
   "Programming Language :: Python :: 3.11",
   "Programming Language :: Python :: 3.12",
   "Programming Language :: Python :: 3.13",
+  "Programming Language :: Python :: 3.14",
   "Programming Language :: Python :: Free Threading :: 3 - Stable",
   "Typing :: Typed",
 ]
-dynamic = ["version"]
-dependencies = ["array-api-compat>=1.12.0,<2"]
+dependencies = ["array-api-compat>=1.13.0,<2"]
 
 [project.urls]
 Homepage = "https://github.com/data-apis/array-api-extra"
 "Bug Tracker" = "https://github.com/data-apis/array-api-extra/issues"
 Changelog = "https://github.com/data-apis/array-api-extra/releases"
 
-# Hatch
-
-[tool.hatch]
-version.path = "src/array_api_extra/__init__.py"
-
-[tool.hatch.build.targets.sdist]
-exclude = ["codecov.yml", "pixi.lock", "RELEASING.md", "renovate.json"]
-
-#  Pixi
+# Pixi
 
 [tool.pixi.workspace]
 channels = ["https://prefix.dev/conda-forge"]
 platforms = ["linux-64", "osx-64", "osx-arm64", "win-64"]
+preview = ["pixi-build"]
+
+### array-api-extra package definition ###
+
+[tool.pixi.package.build.backend]
+name = "pixi-build-python"
+version = "*"
+
+[tool.pixi.package.host-dependencies]
+meson-python = "*"
+uv = "*" # interfaces with meson-python instead of pip
+
+[tool.pixi.package.run-dependencies]
+array-api-compat = "*"
+
+### workspace environments ###
+
+[tool.pixi.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
+
+# 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 ###
+
+[tool.pixi.dev]
+# this pulls in array-api-extra's host and run dependencies
+array-api-extra.path = "."
 
 [tool.pixi.dependencies]
-array-api-compat = ">=1.12.0,<2"
+array-api-extra.path = "."
 
-[tool.pixi.pypi-dependencies]
-array-api-extra = { path = ".", editable = true }
+### non-default feature definitions ###
 
 [tool.pixi.feature.lint.dependencies]
 typing-extensions = ">=4.15.0"
 pylint = ">=4.0.4"
 mypy = ">=1.19.1"
-basedpyright = ">=1.36.1"
+basedpyright = ">=1.38.0"
 numpydoc = ">=1.10.0,<2"
 # import dependencies for mypy:
 array-api-strict = ">=2.4.1,<2.5"
 numpy = ">=2.1.3"
-hypothesis = ">=6.148.2"
-dask-core = ">=2025.12.0" # No distributed, tornado, etc.
+hypothesis = ">=6.151.2"
+dask-core = ">=2026.1.2" # No distributed, tornado, etc.
 dprint = ">=0.50.0,<0.51"
-lefthook = ">=2.0.12,<3"
-ruff = ">=0.14.10,<0.15"
-typos = ">=1.40.0,<2"
-actionlint = ">=1.7.9,<2"
+lefthook = ">=2.1.1,<3"
+ruff = ">=0.15.1,<0.16"
+typos = ">=1.43.4,<2"
+actionlint = ">=1.7.10,<2"
 blacken-docs = ">=1.20.0,<2"
 pytest = ">=9.0.2,<10"
-validate-pyproject = ">=0.24.1,<0.25"
+validate-pyproject = ">=0.25,<0.26"
 # NOTE: don't add cupy, jax, pytorch, or sparse here,
 # as they slow down mypy and are not portable across target OSs
 
@@ -95,9 +131,10 @@ lint = { cmd = "lefthook run pre-commit --all-files --force", description = "Run
 [tool.pixi.feature.tests.dependencies]
 pytest = ">=9.0.2"
 pytest-cov = ">=7.0.0"
-hypothesis = ">=6.148.2"
+hypothesis = ">=6.151.2"
 array-api-strict = ">=2.4.1,<2.5"
 numpy = ">=1.22.0"
+scipy = ">=1.15.2,<2"
 
 [tool.pixi.feature.tests.tasks]
 tests = { cmd = "pytest -v", description = "Run tests" }
@@ -116,11 +153,11 @@ open-coverage = { cmd = "open htmlcov/index.html", depends-on = ["coverage"], de
 [tool.pixi.feature.docs.dependencies]
 sphinx = ">=7.4.7"
 furo = ">=2024.8.6"
-myst-parser = ">=4.0.1"
+myst-parser = ">=5.0.0"
 sphinx-copybutton = ">=0.5.2"
 sphinx-autodoc-typehints = ">=1.25.3"
 # Needed to import parsed modules with autodoc
-dask-core = ">=2025.12.0" # No distributed, tornado, etc.
+dask-core = ">=2026.1.2" # No distributed, tornado, etc.
 pytest = ">=9.0.2"
 typing-extensions = ">=4.15.0"
 numpy = ">=2.1.3"
@@ -135,35 +172,32 @@ ipython = ">=7.33.0"
 [tool.pixi.feature.dev.tasks]
 ipython = { cmd = "ipython", description = "Launch ipython" }
 
-[tool.pixi.feature.py310.dependencies]
-python = "~=3.10.0"
+[tool.pixi.feature.py311.dependencies]
+python = "~=3.11.0"
 
-[tool.pixi.feature.py313.dependencies]
-python = "~=3.13.0"
+[tool.pixi.feature.py314.dependencies]
+python = "~=3.14.0"
 
 [tool.pixi.feature.numpy1.dependencies]
 # Oldest NumPy version supported by scikit-learn.
 # Note that this is older than what SPEC0 recommends.
-numpy = "=1.22.0"
+numpy = "=1.24.1"
 
 # Backends that can run on CPU-only hosts
 # Note: JAX and PyTorch will install CPU variants.
 [tool.pixi.feature.backends.dependencies]
-pytorch = ">=2.7.1"
-dask-core = ">=2025.12.0" # No distributed, tornado, etc.
+pytorch = ">=2.10.0"
+dask-core = ">=2026.1.2" # No distributed, tornado, etc.
 sparse = ">=0.17.0"
 
 [tool.pixi.feature.backends.target.linux-64.dependencies]
-# On CPU Python 3.10, use 0.6.2
-# On CPU Python >=3.11, use >=0.7.0
-# On GPU, use 0.6.0 (0.6.2 and 0.7.0 both segfault); see jaxlib pin below.
-jax = ">=0.6.0"
+jax = ">=0.7.2"
 
 [tool.pixi.feature.backends.target.osx-64.dependencies]
-jax = ">=0.6.0"
+jax = ">=0.7.2"
 
 [tool.pixi.feature.backends.target.osx-arm64.dependencies]
-jax = ">=0.6.0"
+jax = ">=0.7.2"
 
 [tool.pixi.feature.backends.target.win-64.dependencies]
 # jax = "*"  # unavailable
@@ -177,60 +211,30 @@ jax = ">=0.6.0"
 [tool.pixi.feature.cuda-backends]
 system-requirements = { cuda = "12" }
 
-[tool.pixi.feature.cuda-backends.target.linux-64.dependencies]
+[tool.pixi.feature.cuda-backends.target.linux.dependencies]
 cupy = ">=13.6.0"
-# JAX 0.6.2 and 0.7.0 segfault on CUDA
-jaxlib = { version = ">=0.6.0,!=0.6.2,!=0.7.0", build = "cuda12*" }
-pytorch = { version = ">=2.7.1", build = "cuda12*" }
-
-[tool.pixi.feature.cuda-backends.target.osx-64.dependencies]
-# cupy = "*"  # unavailable
-# jaxlib = { version = "*", build = "cuda12*" }  # unavailable
-# pytorch = { version = "*", build = "cuda12*" }  # unavailable
+jaxlib = { version = ">=0.7.2", build = "cuda12*" }
+pytorch = { version = ">=2.10.0", build = "cuda12*" }
 
-[tool.pixi.feature.cuda-backends.target.osx-arm64.dependencies]
+[tool.pixi.feature.cuda-backends.target.osx.dependencies]
 # cupy = "*"  # unavailable
 # jaxlib = { version = "*", build = "cuda12*" }  # unavailable
 # pytorch = { version = "*", build = "cuda12*" }  # unavailable
 
-[tool.pixi.feature.cuda-backends.target.win-64.dependencies]
+[tool.pixi.feature.cuda-backends.target.win.dependencies]
 cupy = ">=13.6.0"
 # jaxlib = { version = "*", build = "cuda12*" }  # unavailable
-pytorch = { version = ">=2.8.0", build = "cuda12*" }
+pytorch = { version = ">=2.10.0", build = "cuda12*" }
 
 [tool.pixi.feature.nogil.dependencies]
 python-freethreading = "~=3.13.0"
-pytest-run-parallel = ">=0.8.0"
+pytest-run-parallel = ">=0.8.2"
 numpy = ">=2.3.5"
 # pytorch = "*"  # Not available on Python 3.13t yet
-dask-core = ">=2025.12.0" # No distributed, tornado, etc.
+dask-core = ">=2026.1.2" # No distributed, tornado, etc.
 # sparse = "*"  # numba not available on Python 3.13t yet
 # jax = "*"  # ml_dtypes not available on Python 3.13t yet
 
-[tool.pixi.environments]
-default = { features = ["py313"], solve-group = "py313" }
-lint = { features = ["py313", "lint"], solve-group = "py313" }
-docs = { features = ["py313", "docs"], solve-group = "py313" }
-tests = { features = ["py313", "tests"], solve-group = "py313" }
-tests-py313 = { features = ["py313", "tests"], solve-group = "py313" } # alias of tests
-
-# Some backends may pin numpy; use separate solve-group
-dev = { features = ["py313", "lint", "tests", "docs", "dev", "backends"], solve-group = "backends" }
-tests-backends = { features = ["py313", "tests", "backends"], solve-group = "backends" }
-# Note: Python 3.10 has already been dropped by some backends (like JAX),
-# so this is testing older versions.
-tests-backends-py310 = { features = ["py310", "tests", "backends"] }
-
-# CUDA not available on free github actions and on some developers' PCs
-dev-cuda = { features = ["py313", "lint", "tests", "docs", "dev", "backends", "cuda-backends"], solve-group = "cuda" }
-tests-cuda = { features = ["py313", "tests", "backends", "cuda-backends"], solve-group = "cuda" }
-tests-cuda-py310 = { features = ["py310", "tests", "backends", "cuda-backends"] }
-
-# Ungrouped environments
-tests-numpy1 = ["py310", "tests", "numpy1"]
-tests-py310 = ["py310", "tests"]
-tests-nogil = ["nogil", "tests"]
-
 # pytest
 
 [tool.pytest.ini_options]
@@ -254,7 +258,7 @@ run.source = ["array_api_extra"]
 
 [tool.mypy]
 files = ["src", "tests"]
-python_version = "3.10"
+python_version = "3.11"
 warn_unused_configs = true
 strict = true
 enable_error_code = ["ignore-without-code", "truthy-bool"]
@@ -273,7 +277,7 @@ disable_error_code = ["no-untyped-def"] # test(...) without -> None
 
 [tool.basedpyright]
 include = ["src", "tests"]
-pythonVersion = "3.10"
+pythonVersion = "3.11"
 pythonPlatform = "All"
 typeCheckingMode = "all"
 
@@ -304,7 +308,7 @@ executionEnvironments = [
 # Ruff
 
 [tool.ruff]
-target-version = "py310"
+target-version = "py311"
 
 [tool.ruff.lint]
 extend-select = [
@@ -349,7 +353,7 @@ ignore = [
 # Pylint
 
 [tool.pylint]
-py-version = "3.10"
+py-version = "3.11"
 reports.output-format = "colorized"
 messages_control.disable = [
   "design", # ignore heavily opinionated design checks

{array_api_extra-0.9.2 → array_api_extra-0.10.1}/src/array_api_extra/__init__.py

@@ -12,6 +12,7 @@ from ._delegation import (
     one_hot,
     pad,
     partition,
+    searchsorted,
     setdiff1d,
     sinc,
     union1d,
@@ -26,7 +27,7 @@ from ._lib._funcs import (
 )
 from ._lib._lazy import lazy_apply
 
-__version__ = "0.9.2"
+__version__ = "0.10.1"
 
 # pylint: disable=duplicate-code
 __all__ = [
@@ -49,6 +50,7 @@ __all__ = [
     "one_hot",
     "pad",
     "partition",
+    "searchsorted",
     "setdiff1d",
     "sinc",
     "union1d",

{array_api_extra-0.9.2 → array_api_extra-0.10.1}/src/array_api_extra/_delegation.py

@@ -27,6 +27,7 @@ __all__ = [
     "nan_to_num",
     "one_hot",
     "pad",
+    "searchsorted",
     "sinc",
 ]
 
@@ -632,6 +633,85 @@ def pad(
     return _funcs.pad(x, pad_width, constant_values=constant_values, xp=xp)
 
 
+def searchsorted(
+    x1: Array,
+    x2: Array,
+    /,
+    *,
+    side: Literal["left", "right"] = "left",
+    xp: ModuleType | None = None,
+) -> Array:
+    """
+    Find indices where elements should be inserted to maintain order.
+
+    Find the indices into a sorted array ``x1`` such that if the elements in ``x2``
+    were inserted before the indices, the resulting array would remain sorted.
+
+    The behavior of this function is similar to that of `array_api.searchsorted`,
+    but it relaxes the requirement that `x1` must be one-dimensional.
+    This function is vectorized, treating slices along the last axis
+    as elements and preceding axes as batch (or "loop") dimensions.
+
+    Parameters
+    ----------
+    x1 : Array
+        Input array. Should have a real-valued data type. Must be sorted in ascending
+        order along the last axis.
+    x2 : Array
+        Array containing search values. Should have a real-valued data type. Must have
+        the same shape as ``x1`` except along the last axis.
+    side : {'left', 'right'}, optional
+        Argument controlling which index is returned if an element of ``x2`` is equal to
+        one or more elements of ``x1``: ``'left'`` returns the index of the first of
+        these elements; ``'right'`` returns the next index after the last of these
+        elements. Default: ``'left'``.
+    xp : array_namespace, optional
+        The standard-compatible namespace for the array arguments. Default: infer.
+
+    Returns
+    -------
+    Array: integer array
+        An array of indices with the same shape as ``x2``.
+
+    Examples
+    --------
+    >>> import array_api_strict as xp
+    >>> import array_api_extra as xpx
+    >>> x = xp.asarray([11, 12, 13, 13, 14, 15])
+    >>> xpx.searchsorted(x, xp.asarray([10, 11.5, 14.5, 16]), xp=xp)
+    Array([0, 1, 5, 6], dtype=array_api_strict.int64)
+    >>> xpx.searchsorted(x, xp.asarray(13), xp=xp)
+    Array(2, dtype=array_api_strict.int64)
+    >>> xpx.searchsorted(x, xp.asarray(13), side='right', xp=xp)
+    Array(4, dtype=array_api_strict.int64)
+
+    `searchsorted` is vectorized along the last axis.
+
+    >>> x1 = xp.asarray([[1., 2., 3., 4.], [5., 6., 7., 8.]])
+    >>> x2 = xp.asarray([[1.1, 3.3], [6.6, 8.8]])
+    >>> xpx.searchsorted(x1, x2, xp=xp)
+    Array([[1, 3],
+           [2, 4]], dtype=array_api_strict.int64)
+    """
+    if xp is None:
+        xp = array_namespace(x1, x2)
+
+    if side not in {"left", "right"}:
+        message = "`side` must be either 'left' or 'right'."
+        raise ValueError(message)
+
+    xp_default_int = _funcs.default_dtype(xp, kind="integral")
+    x2_0d = x2.ndim == 0
+    x1_1d = x1.ndim <= 1
+
+    if x1_1d or is_torch_namespace(xp):
+        x2 = xp.reshape(x2, ()) if (x2_0d and x1_1d) else x2
+        out = xp.searchsorted(x1, x2, side=side)
+        return xp.astype(out, xp_default_int, copy=False)
+
+    return _funcs.searchsorted(x1, x2, side=side, xp=xp)
+
+
 def setdiff1d(
     x1: Array | complex,
     x2: Array | complex,
@@ -1048,6 +1128,15 @@ def union1d(a: Array, b: Array, /, *, xp: ModuleType | None = None) -> Array:
     -------
     Array
         Unique, sorted union of the input arrays.
+
+    See Also
+    --------
+    jax.numpy.union1d : Corresponding function in JAX.
+
+    Notes
+    -----
+    This function is not compatible with `jax.jit`.
+    See the docstring of the corresponding JAX function for more information.
     """
     if xp is None:
         xp = array_namespace(a, b)

{array_api_extra-0.9.2 → array_api_extra-0.10.1}/src/array_api_extra/_lib/_at.py

@@ -21,7 +21,7 @@ from ._utils._typing import Array, SetIndex
 
 if TYPE_CHECKING:  # pragma: no cover
     # TODO import from typing (requires Python >=3.11)
-    from typing_extensions import Self
+    from typing import Self
 
 
 class _AtOp(Enum):

{array_api_extra-0.9.2 → array_api_extra-0.10.1}/src/array_api_extra/_lib/_backends.py

@@ -10,7 +10,7 @@ import pytest
 
 __all__ = ["NUMPY_VERSION", "Backend"]
 
-NUMPY_VERSION = tuple(int(v) for v in np.__version__.split(".")[:3])  # pyright: ignore[reportUnknownArgumentType]
+NUMPY_VERSION = tuple(int(v) for v in np.__version__.split(".")[:3])
 
 
 class Backend(Enum):  # numpydoc ignore=PR02