Skip to content

Commit c63767a

Browse files
committed
🎨 simplified TypeVar use in ndarray.__add__
1 parent e06aec0 commit c63767a

File tree

1 file changed

+11
-17
lines changed

1 file changed

+11
-17
lines changed

‎src/numpy-stubs/__init__.pyi

Lines changed: 11 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -646,7 +646,7 @@ _ShapeT_1nd = TypeVar("_ShapeT_1nd", bound=tuple[int, Unpack[tuple[int, ...]]])
646646
_1NShapeT = TypeVar("_1NShapeT", bound=tuple[L[1], Unpack[tuple[L[1], ...]]]) # (1,) | (1, 1) | (1, 1, 1) | ...
647647

648648
_ScalarT = TypeVar("_ScalarT", bound=generic)
649-
_ScalarOutT = TypeVar("_ScalarOutT", bound=generic)
649+
_SelfScalarT = TypeVar("_SelfScalarT", bound=generic)
650650
_ScalarT_co = TypeVar("_ScalarT_co", bound=generic, default=Any, covariant=True)
651651
_IntScalarT = TypeVar("_IntScalarT", bound=bool_ | integer | object_)
652652
_RealNumberT = TypeVar("_RealNumberT", bound=integer | floating)
@@ -2139,17 +2139,15 @@ class ndarray(_ArrayOrScalarCommon, Generic[_ShapeT_co, _DTypeT_co]):
21392139
@overload
21402140
def __add__(self: NDArray[_ScalarT], x: _nt.Casts[_ScalarT], /) -> NDArray[_ScalarT]: ... # type: ignore[overload-overlap]
21412141
@overload
2142-
def __add__(self: NDArray[_ScalarT], x: _nt.CastsWith[_ScalarT, _ScalarOutT], /) -> NDArray[_ScalarOutT]: ... # type: ignore[overload-overlap]
2142+
def __add__(self: NDArray[_SelfScalarT], x: _nt.CastsWith[_SelfScalarT, _ScalarT], /) -> NDArray[_ScalarT]: ... # type: ignore[overload-overlap]
21432143
@overload
2144-
def __add__(self: _nt.CastsWithBuiltin[_T, _ScalarOutT], x: _nt.SequenceND[_T], /) -> NDArray[_ScalarOutT]: ...
2144+
def __add__(self: _nt.CastsWithBuiltin[_T, _ScalarT], x: _nt.SequenceND[_T], /) -> NDArray[_ScalarT]: ...
21452145
@overload
2146-
def __add__(self: _nt.CastsWithInt[_ScalarOutT], x: _nt.SequenceND[_nt.JustInt], /) -> NDArray[_ScalarOutT]: ...
2146+
def __add__(self: _nt.CastsWithInt[_ScalarT], x: _nt.SequenceND[_nt.JustInt], /) -> NDArray[_ScalarT]: ...
21472147
@overload
2148-
def __add__(self: _nt.CastsWithFloat[_ScalarOutT], x: _nt.SequenceND[_nt.JustFloat], /) -> NDArray[_ScalarOutT]: ...
2148+
def __add__(self: _nt.CastsWithFloat[_ScalarT], x: _nt.SequenceND[_nt.JustFloat], /) -> NDArray[_ScalarT]: ...
21492149
@overload
2150-
def __add__(
2151-
self: _nt.CastsWithComplex[_ScalarOutT], x: _nt.SequenceND[_nt.JustComplex], /
2152-
) -> NDArray[_ScalarOutT]: ...
2150+
def __add__(self: _nt.CastsWithComplex[_ScalarT], x: _nt.SequenceND[_nt.JustComplex], /) -> NDArray[_ScalarT]: ...
21532151
@overload
21542152
def __add__(self: NDArray[datetime64], x: _nt.CoTimeDelta_nd, /) -> NDArray[datetime64]: ...
21552153
@overload
@@ -2165,19 +2163,15 @@ class ndarray(_ArrayOrScalarCommon, Generic[_ShapeT_co, _DTypeT_co]):
21652163
@overload
21662164
def __radd__(self: NDArray[_ScalarT], x: _nt.Casts[_ScalarT], /) -> NDArray[_ScalarT]: ... # type: ignore[overload-overlap]
21672165
@overload
2168-
def __radd__(self: NDArray[_ScalarT], x: _nt.CastsWith[_ScalarT, _ScalarOutT], /) -> NDArray[_ScalarOutT]: ... # type: ignore[overload-overlap]
2166+
def __radd__(self: NDArray[_SelfScalarT], x: _nt.CastsWith[_SelfScalarT, _ScalarT], /) -> NDArray[_ScalarT]: ... # type: ignore[overload-overlap]
21692167
@overload
2170-
def __radd__(self: _nt.CastsWithBuiltin[_T, _ScalarOutT], x: _nt.SequenceND[_T], /) -> NDArray[_ScalarOutT]: ...
2168+
def __radd__(self: _nt.CastsWithBuiltin[_T, _ScalarT], x: _nt.SequenceND[_T], /) -> NDArray[_ScalarT]: ...
21712169
@overload
2172-
def __radd__(self: _nt.CastsWithInt[_ScalarOutT], x: _nt.SequenceND[_nt.JustInt], /) -> NDArray[_ScalarOutT]: ...
2170+
def __radd__(self: _nt.CastsWithInt[_ScalarT], x: _nt.SequenceND[_nt.JustInt], /) -> NDArray[_ScalarT]: ...
21732171
@overload
2174-
def __radd__(
2175-
self: _nt.CastsWithFloat[_ScalarOutT], x: _nt.SequenceND[_nt.JustFloat], /
2176-
) -> NDArray[_ScalarOutT]: ...
2172+
def __radd__(self: _nt.CastsWithFloat[_ScalarT], x: _nt.SequenceND[_nt.JustFloat], /) -> NDArray[_ScalarT]: ...
21772173
@overload
2178-
def __radd__(
2179-
self: _nt.CastsWithComplex[_ScalarOutT], x: _nt.SequenceND[_nt.JustComplex], /
2180-
) -> NDArray[_ScalarOutT]: ...
2174+
def __radd__(self: _nt.CastsWithComplex[_ScalarT], x: _nt.SequenceND[_nt.JustComplex], /) -> NDArray[_ScalarT]: ...
21812175
@overload
21822176
def __radd__(self: NDArray[datetime64], x: _nt.CoTimeDelta_nd, /) -> NDArray[datetime64]: ...
21832177
@overload

0 commit comments

Comments
 (0)