diff options
| author | Aaron Meurer <asmeurer@gmail.com> | 2021-07-16 15:26:42 -0600 |
|---|---|---|
| committer | Aaron Meurer <asmeurer@gmail.com> | 2021-07-16 15:27:24 -0600 |
| commit | 687e2a3b1ee167ae8dc359cf7e92b67e551dbbfe (patch) | |
| tree | d4a3f009b5890d913851c90f081410868030cd84 /numpy | |
| parent | 7c5380d61e31db60f44bdc86047c71f2b1f63458 (diff) | |
| download | numpy-687e2a3b1ee167ae8dc359cf7e92b67e551dbbfe.tar.gz | |
Add type hints to the array API __setitem__
Diffstat (limited to 'numpy')
| -rw-r--r-- | numpy/_array_api/_array_object.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/numpy/_array_api/_array_object.py b/numpy/_array_api/_array_object.py index 455e4fb63..06cb1e925 100644 --- a/numpy/_array_api/_array_object.py +++ b/numpy/_array_api/_array_object.py @@ -521,7 +521,7 @@ class Array: res = self._array.__rshift__(other._array) return self.__class__._new(res) - def __setitem__(self, key, value, /): + def __setitem__(self, key: Union[int, slice, ellipsis, Tuple[Union[int, slice, ellipsis], ...], Array], value: Union[int, float, bool, Array], /) -> Array: """ Performs the operation __setitem__. """ |