From 687e2a3b1ee167ae8dc359cf7e92b67e551dbbfe Mon Sep 17 00:00:00 2001 From: Aaron Meurer Date: Fri, 16 Jul 2021 15:26:42 -0600 Subject: Add type hints to the array API __setitem__ --- numpy/_array_api/_array_object.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'numpy/_array_api') 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__. """ -- cgit v1.2.1