🎨 reformat for line-length of max 120

Author: jorenham

src/_numtype/__init__.pyi

@@ -185,7 +185,11 @@ MArray = TypeAliasType("MArray", np.ma.MaskedArray[_ShapeT, np.dtype[_ScalarT0]]
 MArray0D = TypeAliasType("MArray0D", np.ma.MaskedArray[tuple[()], np.dtype[_ScalarT0]], type_params=(_ScalarT0,))
 MArray1D = TypeAliasType("MArray1D", np.ma.MaskedArray[tuple[int], np.dtype[_ScalarT0]], type_params=(_ScalarT0,))
 MArray2D = TypeAliasType("MArray2D", np.ma.MaskedArray[tuple[int, int], np.dtype[_ScalarT0]], type_params=(_ScalarT0,))
-MArray3D = TypeAliasType("MArray3D", np.ma.MaskedArray[tuple[int, int, int], np.dtype[_ScalarT0]], type_params=(_ScalarT0,))
+MArray3D = TypeAliasType(
+    "MArray3D",
+    np.ma.MaskedArray[tuple[int, int, int], np.dtype[_ScalarT0]],
+    type_params=(_ScalarT0,),
+)
 
 Matrix = TypeAliasType("Matrix", np.matrix[tuple[int, int], np.dtype[_ScalarT0]], type_params=(_ScalarT0,))
 

src/_numtype/_nep50.pyi

@@ -7,7 +7,16 @@ from typing_extensions import TypeAliasType, TypeVar
 
 import numpy as np
 
-__all__ = ["CanCast0D", "CanCastND", "CanNEP50", "CanNEP50Complex", "CanNEP50Float", "MatchND", "PromoteWith", "PromoteWith0D"]
+__all__ = [
+    "CanCast0D",
+    "CanCastND",
+    "CanNEP50",
+    "CanNEP50Complex",
+    "CanNEP50Float",
+    "MatchND",
+    "PromoteWith",
+    "PromoteWith0D",
+]
 
 _T_co = TypeVar("_T_co", covariant=True)
 _BelowT_contra = TypeVar("_BelowT_contra", bound=np.generic, contravariant=True)

src/_numtype/_scalar_co.pyi

@@ -1,4 +1,4 @@
-from typing import TypeAlias
+from typing import Any, TypeAlias
 
 import numpy as np
 
@@ -37,32 +37,32 @@ __all__ = [
 # See https://github.com/numpy/numtype/issues/366 for the scalar promotion table and
 # https://numpy.org/neps/nep-0050-scalar-promotion.html for the official specification.
 
-co_integer8: TypeAlias = integer8 | np.bool
-co_integer16: TypeAlias = integer16 | co_integer8
-co_integer32: TypeAlias = integer32 | co_integer16
-co_integer64: TypeAlias = np.integer | np.bool
-co_integer = co_integer64
-
 co_int8: TypeAlias = np.int8 | np.bool
 co_int16: TypeAlias = np.int16 | co_integer8
 co_int32: TypeAlias = np.int32 | co_integer16
 co_long: TypeAlias = np.long | co_int32
-co_int64: TypeAlias = np.signedinteger | co_integer32
+co_int64: TypeAlias = np.signedinteger[Any] | co_integer32
 
 co_uint8: TypeAlias = np.uint8 | np.bool
 co_uint16: TypeAlias = np.uint16 | co_uint8
 co_uint32: TypeAlias = np.uint32 | co_uint16
 co_ulong: TypeAlias = np.ulong | co_uint32
-co_uint64: TypeAlias = np.unsignedinteger | np.bool
+co_uint64: TypeAlias = np.unsignedinteger[Any] | np.bool
+
+co_integer8: TypeAlias = integer8 | np.bool
+co_integer16: TypeAlias = integer16 | co_integer8
+co_integer32: TypeAlias = integer32 | co_integer16
+co_integer64: TypeAlias = np.integer[Any] | np.bool
+co_integer = co_integer64
 
 co_float16: TypeAlias = np.float16 | co_integer8
 co_float32: TypeAlias = np.float32 | np.float16 | co_integer16
 co_float64: TypeAlias = np.float64 | np.float32 | np.float16 | co_integer64
-co_float: TypeAlias = np.floating | co_integer
+co_float: TypeAlias = np.floating[Any] | co_integer
 
 co_complex64: TypeAlias = inexact32 | number16 | integer8
 co_complex128: TypeAlias = number64 | number32 | number16 | co_integer
-co_complex: TypeAlias = np.number | np.bool
+co_complex: TypeAlias = np.number[Any] | np.bool
 co_number = co_complex
 
 co_timedelta: TypeAlias = np.timedelta64 | co_integer

src/numpy-stubs/@test/static/accept/multiarray.pyi

@@ -147,7 +147,9 @@ assert_type(np.frompyfunc(func12, 1, 2).ntypes, int)
 assert_type(np.frompyfunc(func12, 1, 2).identity, Any | None)
 assert_type(np.frompyfunc(func12, 1, 2).signature, str | None)
 assert_type(np.frompyfunc(func12, 1, 2)(f8, f8), tuple[complex, str])
-assert_type(np.frompyfunc(func12, 1, 2)(AR_f8, f8), tuple[complex | npt.NDArray[np.object_], str | npt.NDArray[np.object_]])
+assert_type(
+    np.frompyfunc(func12, 1, 2)(AR_f8, f8), tuple[complex | npt.NDArray[np.object_], str | npt.NDArray[np.object_]]
+)
 
 assert_type(np.datetime_data("m8[D]"), tuple[str, int])
 assert_type(np.datetime_data(np.datetime64), tuple[str, int])

src/numpy-stubs/@test/static/accept/scalars.pyi

@@ -1,10 +1,8 @@
-from typing import Any, Literal, TypeAlias
+from typing import Any, Literal
 from typing_extensions import Unpack, assert_type
 
 import numpy as np
 
-_1: TypeAlias = Literal[1]  # noqa: PYI042
-
 b: np.bool
 u8: np.uint64
 i8: np.int64
@@ -111,17 +109,17 @@ assert_type(S.flatten(), np.ndarray[tuple[int], np.dtype[np.bytes_]])
 
 assert_type(b.reshape(()), np.bool)
 assert_type(i8.reshape([]), np.int64)
-assert_type(b.reshape(1), np.ndarray[tuple[_1], np.dtype[np.bool]])
-assert_type(i8.reshape(-1), np.ndarray[tuple[_1], np.dtype[np.int64]])
-assert_type(u8.reshape(1, 1), np.ndarray[tuple[_1, _1], np.dtype[np.uint64]])
-assert_type(f8.reshape(1, -1), np.ndarray[tuple[_1, _1], np.dtype[np.float64]])
-assert_type(c16.reshape(1, 1, 1), np.ndarray[tuple[_1, _1, _1], np.dtype[np.complex128]])
-assert_type(U.reshape(1, 1, 1, 1), np.ndarray[tuple[_1, _1, _1, _1], np.dtype[np.str_]])
+assert_type(b.reshape(1), np.ndarray[tuple[int], np.dtype[np.bool]])
+assert_type(i8.reshape(-1), np.ndarray[tuple[int], np.dtype[np.int64]])
+assert_type(u8.reshape(1, 1), np.ndarray[tuple[int, int], np.dtype[np.uint64]])
+assert_type(f8.reshape(1, -1), np.ndarray[tuple[int, int], np.dtype[np.float64]])
+assert_type(c16.reshape(1, 1, 1), np.ndarray[tuple[int, int, int], np.dtype[np.complex128]])
+assert_type(U.reshape(1, 1, 1, 1), np.ndarray[tuple[int, int, int, int], np.dtype[np.str_]])
 assert_type(
     S.reshape(1, 1, 1, 1, 1),
     np.ndarray[
-        # len(shape) >= 5
-        tuple[_1, _1, _1, _1, _1, Unpack[tuple[_1, ...]]],
+        # len(shape) >= 4
+        tuple[int, int, int, int, Unpack[tuple[int, ...]]],
         np.dtype[np.bytes_],
     ],
 )

src/numpy-stubs/_core/_dtype.pyi

@@ -8,7 +8,9 @@ import numpy as np
 _T = TypeVar("_T")
 
 _Kind: TypeAlias = L["u", "i", "c", "f", "b", "V", "O", "M", "m", "S", "U"]
-_Name: TypeAlias = L["uint", "int", "complex", "float", "bool", "void", "object", "datetime", "timedelta", "bytes", "str"]
+_Name: TypeAlias = L[
+    "uint", "int", "complex", "float", "bool", "void", "object", "datetime", "timedelta", "bytes", "str"
+]
 
 @type_check_only
 class _KindToStemType(TypedDict):

src/numpy-stubs/_core/_internal.pyi

@@ -70,6 +70,8 @@ class dummy_ctype(Generic[_T_co]):
     def __mul__(self, other: object, /) -> Self: ...
     def __call__(self, /, *other: object) -> _T_co: ...
 
-def array_ufunc_errmsg_formatter(dummy: object, ufunc: np.ufunc, method: str, *inputs: object, **kwargs: object) -> str: ...
+def array_ufunc_errmsg_formatter(
+    dummy: object, ufunc: np.ufunc, method: str, *inputs: object, **kwargs: object
+) -> str: ...
 def array_function_errmsg_formatter(public_api: Callable[..., object], types: Iterable[str]) -> str: ...
 def npy_ctypes_check(cls: type) -> bool: ...

src/numpy-stubs/_core/_multiarray_umath.pyi

@@ -97,7 +97,9 @@ _ShapeT = TypeVar("_ShapeT", bound=tuple[int, ...])
 
 _ScalarT = TypeVar("_ScalarT", bound=np.generic)
 _NumericT = TypeVar("_NumericT", bound=np.bool | np.number | np.timedelta64 | np.object_)
-_SafeScalarT = TypeVar("_SafeScalarT", bound=np.bool | np.number | np.flexible | np.timedelta64 | np.datetime64)  # no `object_`
+_SafeScalarT = TypeVar(
+    "_SafeScalarT", bound=np.bool | np.number | np.flexible | np.timedelta64 | np.datetime64
+)  # no `object_`
 
 _DTypeT = TypeVar("_DTypeT", bound=np.dtype)
 
@@ -476,7 +478,9 @@ class nditer:
 
     #
     def __enter__(self) -> Self: ...
-    def __exit__(self, cls: type[BaseException] | None, exc: BaseException | None, tb: TracebackType | None, /) -> None: ...
+    def __exit__(
+        self, cls: type[BaseException] | None, exc: BaseException | None, tb: TracebackType | None, /
+    ) -> None: ...
     def close(self) -> None: ...
     def reset(self) -> None: ...
 

src/numpy-stubs/_core/_ufunc_config.pyi

@@ -38,7 +38,9 @@ class errstate:
     ) -> None: ...
     def __call__(self, /, func: _CallableT) -> _CallableT: ...
     def __enter__(self) -> None: ...
-    def __exit__(self, cls: type[BaseException] | None, exc: BaseException | None, tb: TracebackType | None, /) -> None: ...
+    def __exit__(
+        self, cls: type[BaseException] | None, exc: BaseException | None, tb: TracebackType | None, /
+    ) -> None: ...
 
 def seterr(
     all: _ErrKind | None = None,

src/numpy-stubs/_core/defchararray.pyi

@@ -272,7 +272,9 @@ class chararray(np.ndarray[_ShapeT_co, _DTypeT_co]):
     @overload
     def endswith(self: _StrArray, /, suffix: _ToStrND, start: _ToIntND = 0, end: _ToIntND | None = None) -> _BoolND: ...
     @overload
-    def endswith(self: _BytesArray, /, suffix: _ToBytesND, start: _ToIntND = 0, end: _ToIntND | None = None) -> _BoolND: ...
+    def endswith(
+        self: _BytesArray, /, suffix: _ToBytesND, start: _ToIntND = 0, end: _ToIntND | None = None
+    ) -> _BoolND: ...
 
     #
     @overload
@@ -320,7 +322,9 @@ class chararray(np.ndarray[_ShapeT_co, _DTypeT_co]):
     @overload
     def replace(self: _StrArray, /, old: _ToStrND, new: _ToStrND, count: _ToIntND | None = None) -> _StrArray: ...
     @overload
-    def replace(self: _BytesArray, /, old: _ToBytesND, new: _ToBytesND, count: _ToIntND | None = None) -> _BytesArray: ...
+    def replace(
+        self: _BytesArray, /, old: _ToBytesND, new: _ToBytesND, count: _ToIntND | None = None
+    ) -> _BytesArray: ...
 
     #
     @overload
@@ -369,9 +373,13 @@ class chararray(np.ndarray[_ShapeT_co, _DTypeT_co]):
 
     #
     @overload
-    def startswith(self: _StrArray, /, prefix: _ToStrND, start: _ToIntND = 0, end: _ToIntND | None = None) -> _BoolND: ...
+    def startswith(
+        self: _StrArray, /, prefix: _ToStrND, start: _ToIntND = 0, end: _ToIntND | None = None
+    ) -> _BoolND: ...
     @overload
-    def startswith(self: _BytesArray, /, prefix: _ToBytesND, start: _ToIntND = 0, end: _ToIntND | None = None) -> _BoolND: ...
+    def startswith(
+        self: _BytesArray, /, prefix: _ToBytesND, start: _ToIntND = 0, end: _ToIntND | None = None
+    ) -> _BoolND: ...
 
     #
     @overload
@@ -497,8 +505,14 @@ def asarray(  # type: ignore[overload-overlap]
     order: _Order | None = None,
 ) -> _BytesArray: ...
 @overload  # unicode=False
-def asarray(obj: object, itemsize: int | None = None, *, unicode: L[False], order: _Order | None = None) -> _BytesArray: ...
+def asarray(
+    obj: object, itemsize: int | None = None, *, unicode: L[False], order: _Order | None = None
+) -> _BytesArray: ...
 @overload  # unicode=True
-def asarray(obj: object, itemsize: int | None = None, *, unicode: L[True], order: _Order | None = None) -> _StrArray: ...
+def asarray(
+    obj: object, itemsize: int | None = None, *, unicode: L[True], order: _Order | None = None
+) -> _StrArray: ...
 @overload  # falback
-def asarray(obj: object, itemsize: int | None = None, unicode: bool | None = None, order: _Order | None = None) -> _CharArray: ...
+def asarray(
+    obj: object, itemsize: int | None = None, unicode: bool | None = None, order: _Order | None = None
+) -> _CharArray: ...