Skip to content

Commit f9045aa

Browse files
committed
🎨 replace npt.NDArray with _nt.Array in np.matlib
1 parent a533d1a commit f9045aa

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

‎src/numpy-stubs/matlib.pyi

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
from typing import Any, Literal, TypeAlias, TypeVar, overload
22

3+
import _numtype as _nt
34
import numpy as np
45
import numpy.typing as npt
56

@@ -553,11 +554,7 @@ def identity(n: int, dtype: npt.DTypeLike | None = None) -> _Matrix[Any]: ...
553554
#
554555
@overload
555556
def eye(
556-
n: int,
557-
M: int | None = None,
558-
k: int = 0,
559-
dtype: type[np.float64] | None = ...,
560-
order: _Order = "C",
557+
n: int, M: int | None = None, k: int = 0, dtype: type[np.float64] | None = ..., order: _Order = "C"
561558
) -> _Matrix[np.float64]: ...
562559
@overload
563560
def eye(n: int, M: int | None, k: int, dtype: _DTypeLike[_T], order: _Order = "C") -> _Matrix[_T]: ...
@@ -582,6 +579,6 @@ def randn(arg: int, /, *args: int) -> _Matrix[np.float64]: ...
582579
@overload
583580
def repmat(a: _Matrix[_T], m: int, n: int) -> _Matrix[_T]: ...
584581
@overload
585-
def repmat(a: _ArrayLike[_T], m: int, n: int) -> npt.NDArray[_T]: ...
582+
def repmat(a: _ArrayLike[_T], m: int, n: int) -> _nt.Array[_T]: ...
586583
@overload
587-
def repmat(a: npt.ArrayLike, m: int, n: int) -> npt.NDArray[Any]: ...
584+
def repmat(a: npt.ArrayLike, m: int, n: int) -> _nt.Array[Any]: ...

0 commit comments

Comments
 (0)