Add return type annotations to the function stubs

Author: asmeurer

array_api_tests/function_stubs/array_object.py

@@ -16,157 +16,157 @@
 
 from ._types import array
 
-def __abs__(x: array):
+def __abs__(x: array) -> array:
     """
     Note: __abs__ is a method of the array object.
     """
     pass
 
-def __add__(x1: array, x2: array):
+def __add__(x1: array, x2: array) -> array:
     """
     Note: __add__ is a method of the array object.
     """
     pass
 
-def __and__(x1: array, x2: array):
+def __and__(x1: array, x2: array) -> array:
     """
     Note: __and__ is a method of the array object.
     """
     pass
 
-def __eq__(x1: array, x2: array):
+def __eq__(x1: array, x2: array) -> array:
     """
     Note: __eq__ is a method of the array object.
     """
     pass
 
-def __floordiv__(x1: array, x2: array):
+def __floordiv__(x1: array, x2: array) -> array:
     """
     Note: __floordiv__ is a method of the array object.
     """
     pass
 
-def __ge__(x1: array, x2: array):
+def __ge__(x1: array, x2: array) -> array:
     """
     Note: __ge__ is a method of the array object.
     """
     pass
 
-def __ge__(x1: array, x2: array):
+def __getitem__(x, key):
     """
     Note: __getitem__ is a method of the array object.
     """
     pass
 
-def __gt__(x1: array, x2: array):
+def __gt__(x1: array, x2: array) -> array:
     """
     Note: __gt__ is a method of the array object.
     """
     pass
 
-def __invert__(x: array):
+def __invert__(x: array) -> array:
     """
     Note: __invert__ is a method of the array object.
     """
     pass
 
-def __le__(x1: array, x2: array):
+def __le__(x1: array, x2: array) -> array:
     """
     Note: __le__ is a method of the array object.
     """
     pass
 
-def __le__(x1: array, x2: array):
+def __len__(x):
     """
     Note: __len__ is a method of the array object.
     """
     pass
 
-def __lshift__(x1: array, x2: array):
+def __lshift__(x1: array, x2: array) -> array:
     """
     Note: __lshift__ is a method of the array object.
     """
     pass
 
-def __lt__(x1: array, x2: array):
+def __lt__(x1: array, x2: array) -> array:
     """
     Note: __lt__ is a method of the array object.
     """
     pass
 
-def __matmul__(x1: array, x2: array):
+def __matmul__(x1: array, x2: array) -> array:
     """
     Note: __matmul__ is a method of the array object.
     """
     pass
 
-def __mod__(x1: array, x2: array):
+def __mod__(x1: array, x2: array) -> array:
     """
     Note: __mod__ is a method of the array object.
     """
     pass
 
-def __mul__(x1: array, x2: array):
+def __mul__(x1: array, x2: array) -> array:
     """
     Note: __mul__ is a method of the array object.
     """
     pass
 
-def __ne__(x1: array, x2: array):
+def __ne__(x1: array, x2: array) -> array:
     """
     Note: __ne__ is a method of the array object.
     """
     pass
 
-def __neg__(x: array):
+def __neg__(x: array) -> array:
     """
     Note: __neg__ is a method of the array object.
     """
     pass
 
-def __or__(x1: array, x2: array):
+def __or__(x1: array, x2: array) -> array:
     """
     Note: __or__ is a method of the array object.
     """
     pass
 
-def __pos__(x: array):
+def __pos__(x: array) -> array:
     """
     Note: __pos__ is a method of the array object.
     """
     pass
 
-def __pow__(x1: array, x2: array):
+def __pow__(x1: array, x2: array) -> array:
     """
     Note: __pow__ is a method of the array object.
     """
     pass
 
-def __rshift__(x1: array, x2: array):
+def __rshift__(x1: array, x2: array) -> array:
     """
     Note: __rshift__ is a method of the array object.
     """
     pass
 
-def __rshift__(x1: array, x2: array):
+def __setitem__(x, key, value):
     """
     Note: __setitem__ is a method of the array object.
     """
     pass
 
-def __sub__(x1: array, x2: array):
+def __sub__(x1: array, x2: array) -> array:
     """
     Note: __sub__ is a method of the array object.
     """
     pass
 
-def __truediv__(x1: array, x2: array):
+def __truediv__(x1: array, x2: array) -> array:
     """
     Note: __truediv__ is a method of the array object.
     """
     pass
 
-def __xor__(x1: array, x2: array):
+def __xor__(x1: array, x2: array) -> array:
     """
     Note: __xor__ is a method of the array object.
     """

array_api_tests/function_stubs/creation_functions.py

@@ -16,37 +16,37 @@
 
 from ._types import Optional, Tuple, Union, array, device, dtype
 
-def arange(start: Union[int, float], *, stop: Optional[Union[int, float]] = None, step: Union[int, float] = 1, dtype: Optional[dtype] = None, device: Optional[device] = None):
+def arange(start: Union[int, float], *, stop: Optional[Union[int, float]] = None, step: Union[int, float] = 1, dtype: Optional[dtype] = None, device: Optional[device] = None) -> array:
     pass
 
-def empty(shape: Union[int, Tuple[int, ...]], *, dtype: Optional[dtype] = None, device: Optional[device] = None):
+def empty(shape: Union[int, Tuple[int, ...]], *, dtype: Optional[dtype] = None, device: Optional[device] = None) -> array:
     pass
 
-def empty_like(x: array, *, dtype: Optional[dtype] = None, device: Optional[device] = None):
+def empty_like(x: array, *, dtype: Optional[dtype] = None, device: Optional[device] = None) -> array:
     pass
 
-def eye(N: int, *, M: Optional[int] = None, k: Optional[int] = 0, dtype: Optional[dtype] = None, device: Optional[device] = None):
+def eye(N: int, *, M: Optional[int] = None, k: Optional[int] = 0, dtype: Optional[dtype] = None, device: Optional[device] = None) -> array:
     pass
 
-def full(shape: Union[int, Tuple[int, ...]], fill_value: Union[int, float], *, dtype: Optional[dtype] = None, device: Optional[device] = None):
+def full(shape: Union[int, Tuple[int, ...]], fill_value: Union[int, float], *, dtype: Optional[dtype] = None, device: Optional[device] = None) -> array:
     pass
 
-def full_like(x: array, fill_value: Union[int, float], *, dtype: Optional[dtype] = None, device: Optional[device] = None):
+def full_like(x: array, fill_value: Union[int, float], *, dtype: Optional[dtype] = None, device: Optional[device] = None) -> array:
     pass
 
-def linspace(start: Union[int, float], stop: Union[int, float], num: int, *, dtype: Optional[dtype] = None, device: Optional[device] = None, endpoint: Optional[bool] = True):
+def linspace(start: Union[int, float], stop: Union[int, float], num: int, *, dtype: Optional[dtype] = None, device: Optional[device] = None, endpoint: Optional[bool] = True) -> array:
     pass
 
-def ones(shape: Union[int, Tuple[int, ...]], *, dtype: Optional[dtype] = None, device: Optional[device] = None):
+def ones(shape: Union[int, Tuple[int, ...]], *, dtype: Optional[dtype] = None, device: Optional[device] = None) -> array:
     pass
 
-def ones_like(x: array, *, dtype: Optional[dtype] = None, device: Optional[device] = None):
+def ones_like(x: array, *, dtype: Optional[dtype] = None, device: Optional[device] = None) -> array:
     pass
 
-def zeros(shape: Union[int, Tuple[int, ...]], *, dtype: Optional[dtype] = None, device: Optional[device] = None):
+def zeros(shape: Union[int, Tuple[int, ...]], *, dtype: Optional[dtype] = None, device: Optional[device] = None) -> array:
     pass
 
-def zeros_like(x: array, *, dtype: Optional[dtype] = None, device: Optional[device] = None):
+def zeros_like(x: array, *, dtype: Optional[dtype] = None, device: Optional[device] = None) -> array:
     pass
 
 __all__ = ['arange', 'empty', 'empty_like', 'eye', 'full', 'full_like', 'linspace', 'ones', 'ones_like', 'zeros', 'zeros_like']