chore: wip

Author: 34j

pyproject.toml

@@ -43,7 +43,6 @@ dev = [
   "pytest>=8,<9",
   "pytest-cov>=6,<7",
   "ruff>=0.11.13",
-  "ssort>=0.14.0",
 ]
 docs = [
   "furo>=2023.5.20; python_version>='3.11'",

src/array_api/_2022_12.py

@@ -15,6 +15,35 @@
 inf = float("inf")
 
 
+@runtime_checkable
+class NestedSequence[T_t_co](Protocol):
+    def __getitem__(self, key: int, /) -> "Union[T_t_co, NestedSequence[T_t_co]]": ...
+
+    def __len__(self, /) -> int: ...
+
+
+@runtime_checkable
+class iinfo_object[TDtype](Protocol):
+    """Dataclass returned by `iinfo`."""
+
+    bits: int
+    max: int
+    min: int
+    dtype: TDtype
+
+
+@runtime_checkable
+class finfo_object[TDtype](Protocol):
+    """Dataclass returned by `finfo`."""
+
+    bits: int
+    eps: float
+    max: float
+    min: float
+    smallest_normal: float
+    dtype: TDtype
+
+
 @runtime_checkable
 class Array[TPycapsule, TArray: Array, TDevice, TDtype, TEllipsis](Protocol):
     def __init__(self: TArray) -> None:
@@ -1195,18 +1224,6 @@ class can_cast[TArray: Array, TDtype](Protocol):
     def __call__(self, from_: Union[TDtype, TArray], to: TDtype, /) -> bool: ...
 
 
-@runtime_checkable
-class finfo_object[TDtype](Protocol):
-    """Dataclass returned by `finfo`."""
-
-    bits: int
-    eps: float
-    max: float
-    min: float
-    smallest_normal: float
-    dtype: TDtype
-
-
 @runtime_checkable
 class finfo[TArray: Array, TDtype](Protocol):
     """
@@ -1263,16 +1280,6 @@ class finfo[TArray: Array, TDtype](Protocol):
     def __call__(self, type: Union[TDtype, TArray], /) -> finfo_object: ...
 
 
-@runtime_checkable
-class iinfo_object[TDtype](Protocol):
-    """Dataclass returned by `iinfo`."""
-
-    bits: int
-    max: int
-    min: int
-    dtype: TDtype
-
-
 @runtime_checkable
 class iinfo[TArray: Array, TDtype](Protocol):
     """
@@ -1713,13 +1720,6 @@ class arange[TArray: Array, TDevice, TDtype](Protocol):
     def __call__(self, start: Union[int, float], /, stop: Optional[Union[int, float]] = None, step: Union[int, float] = 1, *, dtype: Optional[TDtype] = None, device: Optional[TDevice] = None) -> TArray: ...
 
 
-@runtime_checkable
-class NestedSequence[T_t_co](Protocol):
-    def __getitem__(self, key: int, /) -> "Union[T_t_co, NestedSequence[T_t_co]]": ...
-
-    def __len__(self, /) -> int: ...
-
-
 @runtime_checkable
 class asarray[TSupportsbufferprotocol, TArray: Array, TDevice, TDtype](Protocol):
     r"""
@@ -7383,51 +7383,6 @@ class unique_values[TArray: Array](Protocol):
     def __call__(self, x: TArray, /) -> TArray: ...
 
 
-@runtime_checkable
-class LinalgNamespace[TArray: Array, TDtype](Protocol):
-    cholesky: cholesky[TArray,]
-    cross: cross[TArray,]
-    det: det[TArray,]
-    diagonal: diagonal[TArray,]
-    eigh: eigh[TArray,]
-    eigvalsh: eigvalsh[TArray,]
-    inv: inv[TArray,]
-    matmul: matmul[TArray,]
-    matrix_norm: matrix_norm[TArray,]
-    matrix_power: matrix_power[TArray,]
-    matrix_rank: matrix_rank[TArray,]
-    matrix_transpose: matrix_transpose[TArray,]
-    outer: outer[TArray,]
-    pinv: pinv[TArray,]
-    qr: qr[TArray,]
-    slogdet: slogdet[TArray,]
-    solve: solve[TArray,]
-    svd: svd[TArray,]
-    svdvals: svdvals[TArray,]
-    tensordot: tensordot[TArray,]
-    trace: trace[TArray, TDtype]
-    vecdot: vecdot[TArray,]
-    vector_norm: vector_norm[TArray,]
-
-
-@runtime_checkable
-class FftNamespace[TArray: Array, TDevice](Protocol):
-    fft: fft[TArray,]
-    ifft: ifft[TArray,]
-    fftn: fftn[TArray,]
-    ifftn: ifftn[TArray,]
-    rfft: rfft[TArray,]
-    irfft: irfft[TArray,]
-    rfftn: rfftn[TArray,]
-    irfftn: irfftn[TArray,]
-    hfft: hfft[TArray,]
-    ihfft: ihfft[TArray,]
-    fftfreq: fftfreq[TArray, TDevice]
-    rfftfreq: rfftfreq[TArray, TDevice]
-    fftshift: fftshift[TArray,]
-    ifftshift: ifftshift[TArray,]
-
-
 @runtime_checkable
 class ArrayNamespace[TSupportsbufferprotocol, TArray: Array, TDevice, TDtype](Protocol):
     astype: astype[TArray, TDtype]
@@ -7542,18 +7497,81 @@ class ArrayNamespace[TSupportsbufferprotocol, TArray: Array, TDevice, TDtype](Pr
     squeeze: squeeze[TArray,]
     stack: stack[TArray,]
     e: float
+    "\nIEEE 754 floating-point representation of Euler's constant.\n\n``e = 2.71828182845904523536028747135266249775724709369995...``\n"
     inf: float
+    "\nIEEE 754 floating-point representation of (positive) infinity.\n"
     nan: float
+    "\nIEEE 754 floating-point representation of Not a Number (``NaN``).\n"
     newaxis: float
+    "\nAn alias for ``None`` which is useful for indexing arrays.\n"
     pi: float
+    "\nIEEE 754 floating-point representation of the mathematical constant ``π``.\n\n``pi = 3.1415926535897932384626433...``\n"
     unique_all: unique_all[TArray,]
     unique_counts: unique_counts[TArray,]
     unique_inverse: unique_inverse[TArray,]
     unique_values: unique_values[TArray,]
+    bool: float
+    complex128: float
+    complex64: float
+    float32: float
+    float64: float
+    int16: float
+    int32: float
+    int64: float
+    int8: float
+    uint16: float
+    uint32: float
+    uint64: float
+    uint8: float
+    Device: TDevice
+
+
+@runtime_checkable
+class LinalgNamespace[TArray: Array, TDtype](Protocol):
+    cholesky: cholesky[TArray,]
+    cross: cross[TArray,]
+    det: det[TArray,]
+    diagonal: diagonal[TArray,]
+    eigh: eigh[TArray,]
+    eigvalsh: eigvalsh[TArray,]
+    inv: inv[TArray,]
+    matmul: matmul[TArray,]
+    matrix_norm: matrix_norm[TArray,]
+    matrix_power: matrix_power[TArray,]
+    matrix_rank: matrix_rank[TArray,]
+    matrix_transpose: matrix_transpose[TArray,]
+    outer: outer[TArray,]
+    pinv: pinv[TArray,]
+    qr: qr[TArray,]
+    slogdet: slogdet[TArray,]
+    solve: solve[TArray,]
+    svd: svd[TArray,]
+    svdvals: svdvals[TArray,]
+    tensordot: tensordot[TArray,]
+    trace: trace[TArray, TDtype]
+    vecdot: vecdot[TArray,]
+    vector_norm: vector_norm[TArray,]
+
+
+@runtime_checkable
+class FftNamespace[TArray: Array, TDevice](Protocol):
+    fft: fft[TArray,]
+    ifft: ifft[TArray,]
+    fftn: fftn[TArray,]
+    ifftn: ifftn[TArray,]
+    rfft: rfft[TArray,]
+    irfft: irfft[TArray,]
+    rfftn: rfftn[TArray,]
+    irfftn: irfftn[TArray,]
+    hfft: hfft[TArray,]
+    ihfft: ihfft[TArray,]
+    fftfreq: fftfreq[TArray, TDevice]
+    rfftfreq: rfftfreq[TArray, TDevice]
+    fftshift: fftshift[TArray,]
+    ifftshift: ifftshift[TArray,]
+
+
+@runtime_checkable
+class ArrayNamespaceFull[TSupportsbufferprotocol, TArray: Array, TDevice, TDtype](ArrayNamespace[TSupportsbufferprotocol, TArray, TDevice, TDtype], Protocol):
     linalg: LinalgNamespace[TArray, TDtype]
     fft: FftNamespace[TArray, TDevice]
-    "\nIEEE 754 floating-point representation of Euler's constant.\n\n``e = 2.71828182845904523536028747135266249775724709369995...``\n"
-    "\nIEEE 754 floating-point representation of (positive) infinity.\n"
-    "\nIEEE 754 floating-point representation of Not a Number (``NaN``).\n"
-    "\nAn alias for ``None`` which is useful for indexing arrays.\n"
-    "\nIEEE 754 floating-point representation of the mathematical constant ``π``.\n\n``pi = 3.1415926535897932384626433...``\n"