1
1
from typing import Any , Literal , TypeAlias , TypeVar , overload
2
2
3
+ import _numtype as _nt
3
4
import numpy as np
4
5
import numpy .typing as npt
5
6
@@ -553,11 +554,7 @@ def identity(n: int, dtype: npt.DTypeLike | None = None) -> _Matrix[Any]: ...
553
554
#
554
555
@overload
555
556
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"
561
558
) -> _Matrix [np .float64 ]: ...
562
559
@overload
563
560
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]: ...
582
579
@overload
583
580
def repmat (a : _Matrix [_T ], m : int , n : int ) -> _Matrix [_T ]: ...
584
581
@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 ]: ...
586
583
@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