diff options
| author | BvB93 <43369155+BvB93@users.noreply.github.com> | 2022-09-30 17:22:19 +0200 |
|---|---|---|
| committer | BvB93 <43369155+BvB93@users.noreply.github.com> | 2022-09-30 17:22:19 +0200 |
| commit | 97465164fe448e394a3cea2ef23ed7ebf9cecb3c (patch) | |
| tree | 8b20c1bbba796ab62e34f2ccdbc1ab9c7373095d /numpy | |
| parent | 17051caef30ad3b99291077c95f5ca9e55f0e082 (diff) | |
| download | numpy-97465164fe448e394a3cea2ef23ed7ebf9cecb3c.tar.gz | |
ENH,TYP: Add special casing for `ndarray`-based indexing
Diffstat (limited to 'numpy')
| -rw-r--r-- | numpy/__init__.pyi | 6 | ||||
| -rw-r--r-- | numpy/typing/tests/data/reveal/ndarray_misc.pyi | 4 |
2 files changed, 8 insertions, 2 deletions
diff --git a/numpy/__init__.pyi b/numpy/__init__.pyi index 992ed908a..623334291 100644 --- a/numpy/__init__.pyi +++ b/numpy/__init__.pyi @@ -1549,6 +1549,12 @@ class ndarray(_ArrayOrScalarCommon, Generic[_ShapeType, _DType_co]): ) -> ndarray[_ShapeType2, _DType]: ... @overload + def __getitem__(self, key: ( + NDArray[integer[Any]] + | NDArray[bool_] + | tuple[NDArray[integer[Any]] | NDArray[bool_], ...] + )) -> ndarray[Any, _DType_co]: ... + @overload def __getitem__(self, key: SupportsIndex | tuple[SupportsIndex, ...]) -> Any: ... @overload def __getitem__(self, key: ( diff --git a/numpy/typing/tests/data/reveal/ndarray_misc.pyi b/numpy/typing/tests/data/reveal/ndarray_misc.pyi index 779d0909b..03fea72dc 100644 --- a/numpy/typing/tests/data/reveal/ndarray_misc.pyi +++ b/numpy/typing/tests/data/reveal/ndarray_misc.pyi @@ -200,8 +200,8 @@ reveal_type(AR_f8.__array_wrap__(B)) # E: ndarray[Any, dtype[object_]] reveal_type(AR_V[0]) # E: Any reveal_type(AR_V[0, 0]) # E: Any -reveal_type(AR_V[AR_i8]) # E: Any -reveal_type(AR_V[AR_i8, AR_i8]) # E: Any +reveal_type(AR_V[AR_i8]) # E: ndarray[Any, dtype[void]] +reveal_type(AR_V[AR_i8, AR_i8]) # E: ndarray[Any, dtype[void]] reveal_type(AR_V[AR_i8, None]) # E: ndarray[Any, dtype[void]] reveal_type(AR_V[0, ...]) # E: ndarray[Any, dtype[void]] reveal_type(AR_V[[0]]) # E: ndarray[Any, dtype[void]] |