Skip to content

Commit 9bf3100

Browse files
authored
🩹 fix numpy.inner overload (#270)
1 parent 189db80 commit 9bf3100

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

‎src/numpy-stubs/_core/_multiarray_umath.pyi

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1473,7 +1473,12 @@ def vdot(a: _ArrayLikeFloat_co, b: _ArrayLikeFloat_co, /) -> np.floating | np.si
14731473
@overload
14741474
def vdot(a: _ArrayLikeNumber_co, b: _ArrayLikeNumber_co, /) -> np.inexact | np.signedinteger: ...
14751475

1476-
# TODO(jorenham): https://github.com/numpy/numtype/issues/117
1476+
#
1477+
@overload
1478+
def inner(a: _ScalarT, b: _ScalarT, /) -> _ScalarT: ...
1479+
@overload
1480+
def inner(a: _Array1D[_ScalarT], b: _Array1D[_ScalarT], /) -> _ScalarT: ...
1481+
@overload
14771482
def inner(a: npt.ArrayLike, b: npt.ArrayLike, /) -> Incomplete: ...
14781483

14791484
#

0 commit comments

Comments
 (0)