diff options
| author | Bas van Beek <b.f.van.beek@vu.nl> | 2022-02-17 12:06:56 +0100 |
|---|---|---|
| committer | Bas van Beek <b.f.van.beek@vu.nl> | 2022-02-17 12:06:56 +0100 |
| commit | ff4424d571f40a02076645b98e0d26d3483dd9c5 (patch) | |
| tree | 3aac0c1dc920955e7129d3b8a6168e95524126c5 /numpy | |
| parent | 1168868df63678e5704acd866fafcf40dc849481 (diff) | |
| download | numpy-ff4424d571f40a02076645b98e0d26d3483dd9c5.tar.gz | |
MAINT: Allow only a single axis in `ndarray.argmin/-max`
Diffstat (limited to 'numpy')
| -rw-r--r-- | numpy/__init__.pyi | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/numpy/__init__.pyi b/numpy/__init__.pyi index 80b4047fc..710c97c26 100644 --- a/numpy/__init__.pyi +++ b/numpy/__init__.pyi @@ -1095,7 +1095,7 @@ class _ArrayOrScalarCommon: @overload def argmax( self, - axis: _ShapeLike = ..., + axis: SupportsIndex = ..., out: None = ..., *, keepdims: bool = ..., @@ -1103,7 +1103,7 @@ class _ArrayOrScalarCommon: @overload def argmax( self, - axis: None | _ShapeLike = ..., + axis: None | SupportsIndex = ..., out: _NdArraySubClass = ..., *, keepdims: bool = ..., @@ -1120,7 +1120,7 @@ class _ArrayOrScalarCommon: @overload def argmin( self, - axis: _ShapeLike = ..., + axis: SupportsIndex = ..., out: None = ..., *, keepdims: bool = ..., @@ -1128,7 +1128,7 @@ class _ArrayOrScalarCommon: @overload def argmin( self, - axis: None | _ShapeLike = ..., + axis: None | SupportsIndex = ..., out: _NdArraySubClass = ..., *, keepdims: bool = ..., |