diff options
Diffstat (limited to 'numpy/lib')
| -rw-r--r-- | numpy/lib/npyio.pyi | 20 | ||||
| -rw-r--r-- | numpy/lib/twodim_base.pyi | 13 |
2 files changed, 20 insertions, 13 deletions
diff --git a/numpy/lib/npyio.pyi b/numpy/lib/npyio.pyi index 60684c846..4f11fa807 100644 --- a/numpy/lib/npyio.pyi +++ b/numpy/lib/npyio.pyi @@ -27,7 +27,13 @@ from numpy import ( ) from numpy.ma.mrecords import MaskedRecords -from numpy.typing import ArrayLike, DTypeLike, NDArray, _SupportsDType +from numpy.typing import ( + ArrayLike, + DTypeLike, + NDArray, + _SupportsDType, + _SupportsArrayFunc, +) from numpy.core.multiarray import ( packbits as packbits, @@ -144,7 +150,7 @@ def loadtxt( encoding: None | str = ..., max_rows: None | int = ..., *, - like: None | ArrayLike = ... + like: None | _SupportsArrayFunc = ... ) -> NDArray[float64]: ... @overload def loadtxt( @@ -160,7 +166,7 @@ def loadtxt( encoding: None | str = ..., max_rows: None | int = ..., *, - like: None | ArrayLike = ... + like: None | _SupportsArrayFunc = ... ) -> NDArray[_SCT]: ... @overload def loadtxt( @@ -176,7 +182,7 @@ def loadtxt( encoding: None | str = ..., max_rows: None | int = ..., *, - like: None | ArrayLike = ... + like: None | _SupportsArrayFunc = ... ) -> NDArray[Any]: ... def savetxt( @@ -233,7 +239,7 @@ def genfromtxt( encoding: str = ..., *, ndmin: L[0, 1, 2] = ..., - like: None | ArrayLike = ..., + like: None | _SupportsArrayFunc = ..., ) -> NDArray[float64]: ... @overload def genfromtxt( @@ -262,7 +268,7 @@ def genfromtxt( encoding: str = ..., *, ndmin: L[0, 1, 2] = ..., - like: None | ArrayLike = ..., + like: None | _SupportsArrayFunc = ..., ) -> NDArray[_SCT]: ... @overload def genfromtxt( @@ -291,7 +297,7 @@ def genfromtxt( encoding: str = ..., *, ndmin: L[0, 1, 2] = ..., - like: None | ArrayLike = ..., + like: None | _SupportsArrayFunc = ..., ) -> NDArray[Any]: ... @overload diff --git a/numpy/lib/twodim_base.pyi b/numpy/lib/twodim_base.pyi index 76d7e5a9d..491862408 100644 --- a/numpy/lib/twodim_base.pyi +++ b/numpy/lib/twodim_base.pyi @@ -31,6 +31,7 @@ from numpy.typing import ( NDArray, _FiniteNestedSequence, _SupportsArray, + _SupportsArrayFunc, _ArrayLikeInt_co, _ArrayLikeFloat_co, _ArrayLikeComplex_co, @@ -73,7 +74,7 @@ def eye( dtype: None = ..., order: _OrderCF = ..., *, - like: None | ArrayLike = ..., + like: None | _SupportsArrayFunc = ..., ) -> NDArray[float64]: ... @overload def eye( @@ -83,7 +84,7 @@ def eye( dtype: _DTypeLike[_SCT] = ..., order: _OrderCF = ..., *, - like: None | ArrayLike = ..., + like: None | _SupportsArrayFunc = ..., ) -> NDArray[_SCT]: ... @overload def eye( @@ -93,7 +94,7 @@ def eye( dtype: DTypeLike = ..., order: _OrderCF = ..., *, - like: None | ArrayLike = ..., + like: None | _SupportsArrayFunc = ..., ) -> NDArray[Any]: ... @overload @@ -113,7 +114,7 @@ def tri( k: int = ..., dtype: None = ..., *, - like: None | ArrayLike = ... + like: None | _SupportsArrayFunc = ... ) -> NDArray[float64]: ... @overload def tri( @@ -122,7 +123,7 @@ def tri( k: int = ..., dtype: _DTypeLike[_SCT] = ..., *, - like: None | ArrayLike = ... + like: None | _SupportsArrayFunc = ... ) -> NDArray[_SCT]: ... @overload def tri( @@ -131,7 +132,7 @@ def tri( k: int = ..., dtype: DTypeLike = ..., *, - like: None | ArrayLike = ... + like: None | _SupportsArrayFunc = ... ) -> NDArray[Any]: ... @overload |