Run black on the .py files for the API specification

Author: rgommers

spec/API_specification/array_api/_types.py

@@ -7,18 +7,29 @@
 from __future__ import annotations
 
 from dataclasses import dataclass
-from typing import Any, List, Literal, Optional, Sequence, Tuple, TypeVar, Union, Protocol
+from typing import (
+    Any,
+    List,
+    Literal,
+    Optional,
+    Sequence,
+    Tuple,
+    TypeVar,
+    Union,
+    Protocol,
+)
 from enum import Enum
 
-array = TypeVar('array')
-device = TypeVar('device')
-dtype = TypeVar('dtype')
-SupportsDLPack = TypeVar('SupportsDLPack')
-SupportsBufferProtocol = TypeVar('SupportsBufferProtocol')
-PyCapsule = TypeVar('PyCapsule')
+array = TypeVar("array")
+device = TypeVar("device")
+dtype = TypeVar("dtype")
+SupportsDLPack = TypeVar("SupportsDLPack")
+SupportsBufferProtocol = TypeVar("SupportsBufferProtocol")
+PyCapsule = TypeVar("PyCapsule")
 # ellipsis cannot actually be imported from anywhere, so include a dummy here
 # to keep pyflakes happy. https://github.com/python/typeshed/issues/3556
-ellipsis = TypeVar('ellipsis')
+ellipsis = TypeVar("ellipsis")
+
 
 @dataclass
 class finfo_object:
@@ -28,6 +39,7 @@ class finfo_object:
     min: float
     smallest_normal: float
 
+
 @dataclass
 class iinfo_object:
     bits: int
@@ -37,11 +49,32 @@ class iinfo_object:
 
 _T_co = TypeVar("_T_co", covariant=True)
 
+
 class NestedSequence(Protocol[_T_co]):
-    def __getitem__(self, key: int, /) -> Union[_T_co, NestedSequence[_T_co]]: ...
-    def __len__(self, /) -> int: ...
+    def __getitem__(self, key: int, /) -> Union[_T_co, NestedSequence[_T_co]]:
+        ...
+
+    def __len__(self, /) -> int:
+        ...
 
 
-__all__ = ['Any', 'List', 'Literal', 'NestedSequence', 'Optional',
-'PyCapsule', 'SupportsBufferProtocol', 'SupportsDLPack', 'Tuple', 'Union', 'Sequence',
-'array', 'device', 'dtype', 'ellipsis', 'finfo_object', 'iinfo_object', 'Enum']
+__all__ = [
+    "Any",
+    "List",
+    "Literal",
+    "NestedSequence",
+    "Optional",
+    "PyCapsule",
+    "SupportsBufferProtocol",
+    "SupportsDLPack",
+    "Tuple",
+    "Union",
+    "Sequence",
+    "array",
+    "device",
+    "dtype",
+    "ellipsis",
+    "finfo_object",
+    "iinfo_object",
+    "Enum",
+]

spec/API_specification/array_api/array_object.py

@@ -1,9 +1,20 @@
 from __future__ import annotations
 
-from ._types import (array, dtype as Dtype, device as Device, Optional, Tuple,
-                     Union, Any, PyCapsule, Enum, ellipsis)
-
-class _array():
+from ._types import (
+    array,
+    dtype as Dtype,
+    device as Device,
+    Optional,
+    Tuple,
+    Union,
+    Any,
+    PyCapsule,
+    Enum,
+    ellipsis,
+)
+
+
+class _array:
     def __init__(self: array) -> None:
         """
         Initialize the attributes for the array object class.
@@ -246,7 +257,9 @@ def __and__(self: array, other: Union[int, bool, array], /) -> array:
            Element-wise results must equal the results returned by the equivalent element-wise function :func:`~array_api.bitwise_and`.
         """
 
-    def __array_namespace__(self: array, /, *, api_version: Optional[str] = None) -> Any:
+    def __array_namespace__(
+        self: array, /, *, api_version: Optional[str] = None
+    ) -> Any:
         """
         Returns an object that has all the array API functions on it.
 
@@ -298,9 +311,9 @@ def __complex__(self: array, /) -> complex:
 
         - If ``self`` is ``True``, the result is ``1+0j``.
         - If ``self`` is ``False``, the result is ``0+0j``.
-        
+
         For real-valued floating-point operands,
-        
+
         - If ``self`` is ``NaN``, the result is ``NaN + NaN j``.
         - If ``self`` is ``+infinity``, the result is ``+infinity + 0j``.
         - If ``self`` is ``-infinity``, the result is ``-infinity + 0j``.
@@ -317,7 +330,9 @@ def __complex__(self: array, /) -> complex:
             a Python ``complex`` object representing the single element of the array instance.
         """
 
-    def __dlpack__(self: array, /, *, stream: Optional[Union[int, Any]] = None) -> PyCapsule:
+    def __dlpack__(
+        self: array, /, *, stream: Optional[Union[int, Any]] = None
+    ) -> PyCapsule:
         """
         Exports the array for consumption by :func:`~array_api.from_dlpack` as a DLPack capsule.
 
@@ -559,7 +574,13 @@ def __ge__(self: array, other: Union[int, float, array], /) -> array:
            Element-wise results must equal the results returned by the equivalent element-wise function :func:`~array_api.greater_equal`.
         """
 
-    def __getitem__(self: array, key: Union[int, slice, ellipsis, Tuple[Union[int, slice, ellipsis], ...], array], /) -> array:
+    def __getitem__(
+        self: array,
+        key: Union[
+            int, slice, ellipsis, Tuple[Union[int, slice, ellipsis], ...], array
+        ],
+        /,
+    ) -> array:
         """
         Returns ``self[key]``.
 
@@ -1099,7 +1120,14 @@ def __rshift__(self: array, other: Union[int, array], /) -> array:
            Element-wise results must equal the results returned by the equivalent element-wise function :func:`~array_api.bitwise_right_shift`.
         """
 
-    def __setitem__(self: array, key: Union[int, slice, ellipsis, Tuple[Union[int, slice, ellipsis], ...], array], value: Union[int, float, bool, array], /) -> None:
+    def __setitem__(
+        self: array,
+        key: Union[
+            int, slice, ellipsis, Tuple[Union[int, slice, ellipsis], ...], array
+        ],
+        value: Union[int, float, bool, array],
+        /,
+    ) -> None:
         """
         Sets ``self[key]`` to ``value``.
 
@@ -1196,17 +1224,17 @@ def __truediv__(self: array, other: Union[int, float, array], /) -> array:
         +------------+----------------+-----------------+--------------------------+
 
         In general, for complex floating-point operands, real-valued floating-point special cases must independently apply to the real and imaginary component operations involving real numbers as described in the above table.
-        
+
         When ``a``, ``b``, ``c``, or ``d`` are all finite numbers (i.e., a value other than ``NaN``, ``+infinity``, or ``-infinity``), division of complex floating-point operands should be computed as if calculated according to the textbook formula for complex number division
-       
+
         .. math::
            \frac{a + bj}{c + dj} = \frac{(ac + bd) + (bc - ad)j}{c^2 + d^2}
-        
+
         When at least one of ``a``, ``b``, ``c``, or ``d`` is ``NaN``, ``+infinity``, or ``-infinity``,
-        
+
         - If ``a``, ``b``, ``c``, and ``d`` are all ``NaN``, the result is ``NaN + NaN j``.
         - In the remaining cases, the result is implementation dependent.
-        
+
         .. note::
            For complex floating-point operands, the results of special cases may be implementation dependent depending on how an implementation chooses to model complex numbers and complex infinity (e.g., complex plane versus Riemann sphere). For those implementations following C99 and its one-infinity model, when at least one component is infinite, even if the other component is ``NaN``, the complex value is infinite, and the usual arithmetic rules do not apply to complex-complex division. In the interest of performance, other implementations may want to avoid the complex branching logic necessary to implement the one-infinity model and choose to implement all complex-complex division according to the textbook formula. Accordingly, special case behavior is unlikely to be consistent across implementations.
 
@@ -1248,7 +1276,9 @@ def __xor__(self: array, other: Union[int, bool, array], /) -> array:
            Element-wise results must equal the results returned by the equivalent element-wise function :func:`~array_api.bitwise_xor`.
         """
 
-    def to_device(self: array, device: Device, /, *, stream: Optional[Union[int, Any]] = None) -> array:
+    def to_device(
+        self: array, device: Device, /, *, stream: Optional[Union[int, Any]] = None
+    ) -> array:
         """
         Copy the array from the device on which it currently resides to the specified ``device``.
 
@@ -1271,6 +1301,7 @@ def to_device(self: array, device: Device, /, *, stream: Optional[Union[int, Any
            If ``stream`` is given, the copy operation should be enqueued on the provided ``stream``; otherwise, the copy operation should be enqueued on the default stream/queue. Whether the copy is performed synchronously or asynchronously is implementation-dependent. Accordingly, if synchronization is required to guarantee data safety, this must be clearly explained in a conforming library's documentation.
         """
 
+
 array = _array
 
-__all__ = ['array']
+__all__ = ["array"]

spec/API_specification/array_api/constants.py

@@ -5,12 +5,12 @@
 ``e = 2.71828182845904523536028747135266249775724709369995...``
 """
 
-inf = float('inf')
+inf = float("inf")
 """
 IEEE 754 floating-point representation of (positive) infinity.
 """
 
-nan = float('nan')
+nan = float("nan")
 """
 IEEE 754 floating-point representation of Not a Number (``NaN``).
 """
@@ -27,4 +27,4 @@
 ``pi = 3.1415926535897932384626433...``
 """
 
-__all__ = ['e', 'inf', 'nan', 'newaxis', 'pi']
+__all__ = ["e", "inf", "nan", "newaxis", "pi"]