array-api-strict 2.4.1__tar.gz → 2.6__tar.gz

{array_api_strict-2.4.1 → array_api_strict-2.6}/.github/dependabot.yml

@@ -5,6 +5,8 @@ updates:
     directory: "/"
     schedule:
       interval: "weekly"
+    cooldown:
+      default-days: 7     # optional
     groups:
       actions:
         patterns:

{array_api_strict-2.4.1 → array_api_strict-2.6}/.github/workflows/array-api-tests.yml

@@ -4,36 +4,38 @@ on: [push, pull_request]
 
 env:
   PYTEST_ARGS: "-v -rxXfE --hypothesis-disable-deadline --max-examples 500 -n 4"
-  API_VERSIONS: "2023.12 2024.12"
+  API_VERSIONS: "2023.12 2024.12 2025.12"
 
 jobs:
   array-api-tests:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ['3.10', '3.11', '3.12', '3.13']
-        numpy-version: ['1.26', '2.3', 'dev']
+        python-version: ['3.10', '3.13', '3.14']
+        numpy-version: ['1.26', 'latest', 'dev']
         exclude:
           - python-version: '3.10'
-            numpy-version: '2.3'
+            numpy-version: 'latest'
           - python-version: '3.10'
             numpy-version: 'dev'            
           - python-version: '3.13'
             numpy-version: '1.26'
+          - python-version: '3.14'
+            numpy-version: '1.26'
       fail-fast: false
     steps:
     - name: Checkout array-api-strict
-      uses: actions/checkout@v4
+      uses: actions/checkout@v7.0.0
       with:
         path: array-api-strict
     - name: Checkout array-api-tests
-      uses: actions/checkout@v4
+      uses: actions/checkout@v7.0.0
       with:
         repository: data-apis/array-api-tests
         submodules: 'true'
         path: array-api-tests
     - name: Set up Python ${{ matrix.python-version }}
-      uses: actions/setup-python@v5
+      uses: actions/setup-python@v6.3.0
       with:
         python-version: ${{ matrix.python-version }}
     - name: Install Dependencies
@@ -41,6 +43,8 @@ jobs:
         python -m pip install --upgrade pip
         if [[ "${{ matrix.numpy-version }}" == "dev" ]]; then
           python -m pip install --pre --extra-index https://pypi.anaconda.org/scientific-python-nightly-wheels/simple numpy;
+        elif [[ "${{ matrix.numpy-version }}" == "latest" ]]; then
+          python -m pip install numpy
         else
           python -m pip install 'numpy=='${{ matrix.numpy-version }};
         fi

{array_api_strict-2.4.1 → array_api_strict-2.6}/.github/workflows/dependabot-auto-merge.yml

@@ -13,7 +13,7 @@ jobs:
     steps:
       - name: Dependabot metadata
         id: metadata
-        uses: dependabot/fetch-metadata@v2
+        uses: dependabot/fetch-metadata@v3
         with:
           github-token: "${{ secrets.GITHUB_TOKEN }}"
       - name: Enable auto-merge for Dependabot PRs

{array_api_strict-2.4.1 → array_api_strict-2.6}/.github/workflows/docs-build.yml

@@ -6,8 +6,8 @@ jobs:
   docs-build:
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v4
-      - uses: actions/setup-python@v5
+      - uses: actions/checkout@v7.0.0
+      - uses: actions/setup-python@v6.3.0
       - name: Install Dependencies
         run: |
           python -m pip install -r docs/requirements.txt
@@ -16,7 +16,7 @@ jobs:
           cd docs
           make html
       - name: Upload Artifact
-        uses: actions/upload-artifact@v4
+        uses: actions/upload-artifact@v7
         with:
           name: docs-build
           path: docs/_build/html

{array_api_strict-2.4.1 → array_api_strict-2.6}/.github/workflows/docs-deploy.yml

@@ -11,9 +11,9 @@ jobs:
     environment:
       name: docs-deploy
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v7.0.0
       - name: Download Artifact
-        uses: dawidd6/action-download-artifact@v11
+        uses: dawidd6/action-download-artifact@v21
         with:
           workflow: docs-build.yml
           name: docs-build

{array_api_strict-2.4.1 → array_api_strict-2.6}/.github/workflows/publish-package.yml

@@ -30,12 +30,12 @@ jobs:
     runs-on: ubuntu-latest
 
     steps:
-    - uses: actions/checkout@v4
+    - uses: actions/checkout@v7.0.0
       with:
         fetch-depth: 0
 
     - name: Set up Python
-      uses: actions/setup-python@v5
+      uses: actions/setup-python@v6.3.0
       with:
         python-version: '3.x'
 
@@ -62,7 +62,7 @@ jobs:
       run: python -m zipfile --list dist/array_api_strict-*.whl
 
     - name: Upload distribution artifact
-      uses: actions/upload-artifact@v4
+      uses: actions/upload-artifact@v7
       with:
         name: dist-artifact
         path: dist
@@ -83,7 +83,7 @@ jobs:
 
     steps:
     - name: Download distribution artifact
-      uses: actions/download-artifact@v4
+      uses: actions/download-artifact@v8
       with:
         name: dist-artifact
         path: dist
@@ -97,19 +97,19 @@ jobs:
     #   if: >-
     #     (github.event_name == 'push' && startsWith(github.ref, 'refs/tags'))
     #     || (github.event_name == 'workflow_dispatch' && github.event.inputs.publish == 'true')
-    #   uses: pypa/gh-action-pypi-publish@v1.12.4
+    #   uses: pypa/gh-action-pypi-publish@v1.14.0
     #   with:
     #     repository-url: https://test.pypi.org/legacy/
     #     print-hash: true
 
     - name: Publish distribution 📦 to PyPI
       if: github.event_name == 'push' && startsWith(github.ref, 'refs/tags')
-      uses: pypa/gh-action-pypi-publish@v1.12.4
+      uses: pypa/gh-action-pypi-publish@v1.14.0
       with:
         print-hash: true
 
     - name: Create GitHub Release from a Tag
-      uses: softprops/action-gh-release@v2
+      uses: softprops/action-gh-release@v3
       if: startsWith(github.ref, 'refs/tags/')
       with:
           files: dist/*

{array_api_strict-2.4.1 → array_api_strict-2.6}/.github/workflows/ruff.yml

@@ -5,9 +5,9 @@ jobs:
     runs-on: ubuntu-latest
     continue-on-error: true
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v7.0.0
       - name: Install Python
-        uses: actions/setup-python@v5
+        uses: actions/setup-python@v6.3.0
         with:
           python-version: "3.11"
       - name: Install dependencies

{array_api_strict-2.4.1 → array_api_strict-2.6}/.github/workflows/tests.yml

@@ -5,19 +5,21 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ['3.10', '3.11', '3.12', '3.13']
-        numpy-version: ['1.26', '2.3', 'dev']
+        python-version: ['3.10', '3.13', '3.14']
+        numpy-version: ['1.26', 'latest', 'dev']
         exclude:
           - python-version: '3.10'
-            numpy-version: '2.3'
+            numpy-version: 'latest'
           - python-version: '3.10'
             numpy-version: 'dev'            
           - python-version: '3.13'
             numpy-version: '1.26'
+          - python-version: '3.14'
+            numpy-version: '1.26'
       fail-fast: false
     steps:
-      - uses: actions/checkout@v4
-      - uses: actions/setup-python@v5
+      - uses: actions/checkout@v7.0.0
+      - uses: actions/setup-python@v6.3.0
         with:
           python-version: ${{ matrix.python-version }}
       - name: Install Dependencies
@@ -25,10 +27,13 @@ jobs:
           python -m pip install --upgrade pip
           if [[ "${{ matrix.numpy-version }}" == "dev" ]]; then
             python -m pip install --pre --extra-index https://pypi.anaconda.org/scientific-python-nightly-wheels/simple numpy;
+          elif [[ "${{ matrix.numpy-version }}" == "latest" ]]; then
+            python -m pip install numpy
           else
-            python -m pip install 'numpy>=1.26,<2.0';
+            python -m pip install 'numpy=='${{ matrix.numpy-version }}
           fi
-          python -m pip install -r requirements-dev.txt
+          python -m pip install pytest hypothesis
+          python -c'import numpy as np; print(f"{np.__version__ = }")'
       - name: Run Tests
         run: |
           pytest

{array_api_strict-2.4.1 → array_api_strict-2.6}/CHANGELOG.md

@@ -1,5 +1,107 @@
 # Changelog
 
+## 2.6 (2026-07-09)
+
+- `__setitem__` method of array objects got stricter:
+
+   - `__setitem__` method now disallows device transfers: if the right-hand side
+     argument is an array on a device which differs from the device of the left-hand
+     argument, an error is raised. Note that this change follows
+     the draft of a future Array API standard revision, see
+     [gh-1005](https://github.com/data-apis/array-api/pull/1005) for discussion.
+   - `None` is no longer allowed as an indexing key. 
+
+- Array objects expose a DLPack device number, via their `__dlpack_device__` method.
+- `from_dlpack` function now supports `copy=` argument.
+
+- `Device` objects are now allowed to only support a subset of data types, and can
+  have a device-dependent default data types. Note that this enhancement follows
+  the draft of a future Array API standard revision, see
+  [gh-1005](https://github.com/data-apis/array-api/pull/1005) for discussion.
+  - A new device object, `Device("no_float64")` was added which only supports
+    single-precision float and complex data types; this device mimics the behavior
+    of a "metal"/MPS device which does not support double precision floats.
+    Attempting to create an array with float64 or complex128 data type on this device
+    raises an error.
+  - The existing `Device("device2")` object has been modified: it supports float64 data
+    type but defaults to float32 and complex64:
+    `asarray([1.0], device=Device("device2")).dtype` is `float32`.
+
+
+### Contributors
+
+The following users contributed to this release:
+
+Evgeni Burovski,
+Lucas Colley,
+Ralf Gommers,
+Tim Head.
+
+
+## 2.5 (2026-02-23)
+
+This release targets the 2025.12 revision of the Array API standard, and supports
+Python versions 3.10--3.14.
+
+### Major Changes
+
+- The default version of the array API standard is now 2025.12. Previous versions can
+  still be enabled via the [flags API](array-api-strict-flags), e.g., by calling
+  `set_array_api_strict_flags(api_version='2024.12')` or setting
+  `ARRAY_API_STRICT_API_VERSION=2024.12`.
+
+  Note that this support is still relatively undertested. Please [report any
+  issues](https://github.com/data-apis/array-api-strict/issues) you find.
+
+- The following functions, new in the 2025.12 revision, are implemented:
+
+  - `isin`
+  - `broadcast_shapes`
+  - `linalg.eig`
+  - `linalg.eigvals`
+
+The following updates to existing APIs have been implemented, per the 2025.12 Array API
+standard revision:
+
+  - `searchsorted` now accepts a python scalar for its second argument,
+    `x2`.
+  - `expand_dims` now accepts tuples of axes.
+  - `permute_dims` now accepts negative axes, with the usual meaning (`axis=-1` is
+    the last axis, `-2` is the second-to-last axis, and so on).
+
+The following functions now return tuples instead of lists:
+
+  - `__array_namespace_info__().devices`
+  - `broadcast_arrays`
+  - `meshgrid` 
+
+
+- Array objects can now be pickled.
+
+
+### Minor Changes
+
+- Several functions were diverging from the standard, and have been fixed to follow
+  the 2025.12 revision of the standard:
+
+  - `clip` has been updated to accept python `int` scalars for its `min` and `max`
+     arguments for clipping a floating-point array.
+  - `clip` has been updated to always returns a copy of an input array.
+  - `full` and `full_like` now reject array values for the `fill_value` argument.
+  - `expand_dims` accepts its axis argument as either positional or keyword argument.
+
+- `sign` of a complex zero, `sign(0 + 0j)` has been fixed to return zero.
+
+
+### Contributors
+
+The following users contributed to this release:
+
+Evgeni Burovski,
+Lucas Colley,
+Lucy Liu
+
+
 ## 2.4.1 (2025-07-27)
 
 ### Major Changes
@@ -8,7 +110,6 @@
 
 - Support for Python versions 3.10 and 3.11 has been reinstated.
 
-
 ### Minor Changes
 
 - Arithmetic operations no longer accept NumPy arrays.

{array_api_strict-2.4.1/array_api_strict.egg-info → array_api_strict-2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: array_api_strict
-Version: 2.4.1
+Version: 2.6
 Summary: A strict, minimal implementation of the Python array API standard.
 Author: Consortium for Python Data API Standards
 License: Copyright (c) 2021-2024, NumPy Developers, Consortium for Python Data API Standards
@@ -48,6 +48,9 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: numpy
+Provides-Extra: test
+Requires-Dist: pytest; extra == "test"
+Requires-Dist: hypothesis; extra == "test"
 
 # array-api-strict
 

{array_api_strict-2.4.1 → array_api_strict-2.6}/array-api-tests-xfails.txt

@@ -29,3 +29,26 @@ array_api_tests/test_statistical_functions.py::test_sum
 
 array_api_tests/test_special_cases.py::test_nan_propagation[cumulative_prod]
 
+# complex special cases which failed "forever"
+array_api_tests/test_special_cases.py::test_unary[expm1(real(x_i) is +infinity and imag(x_i) is +0) -> +infinity + 0j]
+array_api_tests/test_special_cases.py::test_unary[expm1(real(x_i) is -infinity and imag(x_i) is +infinity) -> -1 + 0j]
+array_api_tests/test_special_cases.py::test_unary[expm1(real(x_i) is +infinity and imag(x_i) is +infinity) -> infinity + NaN j]
+array_api_tests/test_special_cases.py::test_unary[expm1(real(x_i) is -infinity and imag(x_i) is NaN) -> -1 + 0j]
+array_api_tests/test_special_cases.py::test_unary[expm1(real(x_i) is +infinity and imag(x_i) is NaN) -> infinity + NaN j]
+array_api_tests/test_special_cases.py::test_unary[expm1(real(x_i) is NaN and imag(x_i) is +0) -> NaN + 0j]
+array_api_tests/test_special_cases.py::test_unary[expm1((real(x_i) is +0 or real(x_i) == -0) and imag(x_i) is +0) -> 0 + 0j]
+
+array_api_tests/test_special_cases.py::test_unary[sign((real(x_i) is -0 or real(x_i) == +0) and (imag(x_i) is -0 or imag(x_i) == +0)) -> 0 + 0j]
+array_api_tests/test_special_cases.py::test_unary[tanh(real(x_i) is +infinity and isfinite(imag(x_i)) and imag(x_i) > 0) -> 1 + 0j]
+
+# this acosh failure is only seen with python==3.10 and numpy==2.2.6, and not e.g. python 3.12 & numpy 2.4.1
+array_api_tests/test_special_cases.py::test_unary[acosh(real(x_i) is +0 and imag(x_i) is NaN) -> NaN \xb1 \u03c0j/2]
+
+array_api_tests/test_special_cases.py::test_unary[sqrt(real(x_i) is +infinity and isfinite(imag(x_i)) and imag(x_i) > 0) -> +0 + infinity j]
+
+# remove this when array-api-strict 2.6 is released: this is only for array-api-tests,
+# which runs self-tests against a released array-api-strict
+array_api_tests/test_dlpack.py::test_from_dlpack
+
+# NumPy 1.26 : NotImplementedError: The copy argument to __dlpack__ is not yet implemented
+array_api_tests/test_dlpack.py::test_dunder_dlpack

{array_api_strict-2.4.1 → array_api_strict-2.6}/array_api_strict/__init__.py

@@ -73,6 +73,7 @@ __all__ += [
 from ._data_type_functions import (
     astype,
     broadcast_arrays,
+    broadcast_shapes,
     broadcast_to,
     can_cast,
     finfo,
@@ -84,6 +85,7 @@ from ._data_type_functions import (
 __all__ += [
     "astype",
     "broadcast_arrays",
+    "broadcast_shapes",
     "broadcast_to",
     "can_cast",
     "finfo",
@@ -299,9 +301,9 @@ from ._searching_functions import argmax, argmin, nonzero, count_nonzero, search
 
 __all__ += ["argmax", "argmin", "nonzero", "count_nonzero", "searchsorted", "where"]
 
-from ._set_functions import unique_all, unique_counts, unique_inverse, unique_values
+from ._set_functions import unique_all, unique_counts, unique_inverse, unique_values, isin
 
-__all__ += ["unique_all", "unique_counts", "unique_inverse", "unique_values"]
+__all__ += ["unique_all", "unique_counts", "unique_inverse", "unique_values", "isin"]
 
 from ._sorting_functions import argsort, sort
 

{array_api_strict-2.4.1 → array_api_strict-2.6}/array_api_strict/_array_object.py

@@ -20,7 +20,7 @@ import sys
 from collections.abc import Iterator
 from enum import IntEnum
 from types import EllipsisType, ModuleType
-from typing import Any, Final, Literal, SupportsIndex
+from typing import Any, Literal, SupportsIndex, Callable
 
 import numpy as np
 import numpy.typing as npt
@@ -40,35 +40,14 @@ from ._dtypes import (
     _real_to_complex_map,
     _result_type,
 )
+from ._devices import (
+    CPU_DEVICE, Device, device_supports_dtype, _normalize_dl_device, _DLPACK_DEVICE_FOR,
+    DLDeviceType
+)
 from ._flags import get_array_api_strict_flags, set_array_api_strict_flags
 from ._typing import PyCapsule
 
 
-class Device:
-    _device: Final[str]
-    __slots__ = ("_device", "__weakref__")
-
-    def __init__(self, device: str = "CPU_DEVICE"):
-        if device not in ("CPU_DEVICE", "device1", "device2"):
-            raise ValueError(f"The device '{device}' is not a valid choice.")
-        self._device = device
-
-    def __repr__(self) -> str:
-        return f"array_api_strict.Device('{self._device}')"
-
-    def __eq__(self, other: object) -> bool:
-        if not isinstance(other, Device):
-            return False
-        return self._device == other._device
-
-    def __hash__(self) -> int:
-        return hash(("Device", self._device))
-
-
-CPU_DEVICE = Device()
-ALL_DEVICES = (CPU_DEVICE, Device("device1"), Device("device2"))
-
-
 class Array:
     """
     n-d array object for the array API namespace.
@@ -113,10 +92,15 @@ class Array:
             raise TypeError(
                 f"The array_api_strict namespace does not support the dtype '{x.dtype}'"
             )
-        obj._array = x
-        obj._dtype = _dtype
+
         if device is None:
             device = CPU_DEVICE
+        if not device_supports_dtype(device, _dtype):
+            raise ValueError(f"Device {device!r} does not support dtype={_dtype!r}.")
+
+        obj._array = x
+        obj._dtype = _dtype
+
         obj._device = device
         return obj
 
@@ -125,6 +109,9 @@ class Array:
         raise TypeError(
             "The array_api_strict Array object should not be instantiated directly. Use an array creation function, such as asarray(), instead."
         )
+    
+    def __reduce__(self) -> tuple[Callable, tuple[npt.NDArray[Any], Device]]:
+        return (self._new, (self._array, self._device))
 
     # These functions are not required by the spec, but are implemented for
     # the sake of usability.
@@ -397,7 +384,7 @@ class Array:
                 isinstance(i, SupportsIndex)  # i.e. ints
                 or isinstance(i, slice)
                 or i == Ellipsis
-                or i is None
+                or (op == "getitem" and i is None) # `None` disallowed in setitem
                 or isinstance(i, Array)
                 or isinstance(i, np.ndarray)
             ):
@@ -627,22 +614,40 @@ class Array:
                 raise NotImplementedError("The copy argument to __dlpack__ is not yet implemented")
 
             return self._array.__dlpack__(stream=stream)
-        else:
-            kwargs = {'stream': stream}
-            if max_version is not _undef:
-                kwargs['max_version'] = max_version
-            if dl_device is not _undef:
-                kwargs['dl_device'] = dl_device
-            if copy is not _undef:
-                kwargs['copy'] = copy
-            return self._array.__dlpack__(**kwargs)
+
+        kwargs: dict[str, Any] = {'stream': stream}
+        self_dl_device = _normalize_dl_device(*_DLPACK_DEVICE_FOR[self._device])
+        cpu_dl_device = _normalize_dl_device(DLDeviceType.kDLCPU, 0)
+        numpy_dl_device: tuple[IntEnum, int] | None = None
+
+        if dl_device not in [_undef, None]:
+            requested = _normalize_dl_device(dl_device[0], dl_device[1])
+            if requested == self_dl_device:
+                pass
+            elif requested == cpu_dl_device and self_dl_device != cpu_dl_device:
+                if copy is False:
+                    raise BufferError(
+                        "Cannot export array to CPU without copying when copy=False"
+                    )
+                if copy is _undef:
+                    copy = True
+                numpy_dl_device = (DLDeviceType.kDLCPU, 0)
+            else:
+                raise BufferError("unsupported device requested")
+
+        if max_version is not _undef:
+            kwargs['max_version'] = max_version
+        if numpy_dl_device is not None:
+            kwargs['dl_device'] = numpy_dl_device
+        if copy is not _undef:
+            kwargs['copy'] = copy
+        return self._array.__dlpack__(**kwargs)
 
     def __dlpack_device__(self) -> tuple[IntEnum, int]:
         """
         Performs the operation __dlpack_device__.
         """
-        # Note: device support is required for this
-        return self._array.__dlpack_device__()
+        return _DLPACK_DEVICE_FOR[self._device]
 
     def __eq__(self, other: Array | complex, /) -> Array:  # type: ignore[override]
         """
@@ -957,6 +962,10 @@ class Array:
         other = value
         if isinstance(value, (bool, int, float, complex)):
             other = self._promote_scalar(value)
+        else:
+            if value.device != self.device:
+                raise ValueError(f"mismatched devices: {self.device = } != {value.device =}.")
+
         dt = _result_type(self.dtype, other.dtype)
         if dt != self.dtype:
             raise TypeError(f"mismatched dtypes: {self.dtype = } and {other.dtype = }")