Update the function stubs from the spec

Author: asmeurer

array_api_tests/function_stubs/array_object.py

@@ -181,7 +181,7 @@ def __rshift__(self: array, other: Union[int, array], /) -> array:
     """
     pass
 
-def __setitem__(self, key, value, /):
+def __setitem__(self: array, key: Union[int, slice, ellipsis, Tuple[Union[int, slice, ellipsis], ...], array], value: Union[int, float, bool, array], /) -> None:
     """
     Note: __setitem__ is a method of the array object.
     """

array_api_tests/function_stubs/creation_functions.py

@@ -17,7 +17,7 @@
 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 asarray(obj: Union[array, float, NestedSequence[bool|int|float], SupportsDLPack, SupportsBufferProtocol], /, *, dtype: Optional[dtype] = None, device: Optional[device] = None, copy: Optional[bool] = None) -> array:
+def asarray(obj: Union[array, bool, int, float, NestedSequence[bool|int|float], SupportsDLPack, SupportsBufferProtocol], /, *, dtype: Optional[dtype] = None, device: Optional[device] = None, copy: Optional[bool] = None) -> array:
     pass
 
 def empty(shape: Union[int, Tuple[int, ...]], *, dtype: Optional[dtype] = None, device: Optional[device] = None) -> array:

array_api_tests/function_stubs/searching_functions.py

@@ -10,12 +10,12 @@
 
 from __future__ import annotations
 
-from ._types import Tuple, array
+from ._types import Optional, Tuple, array
 
-def argmax(x: array, /, *, axis: int = None, keepdims: bool = False) -> array:
+def argmax(x: array, /, *, axis: Optional[int] = None, keepdims: bool = False) -> array:
     pass
 
-def argmin(x: array, /, *, axis: int = None, keepdims: bool = False) -> array:
+def argmin(x: array, /, *, axis: Optional[int] = None, keepdims: bool = False) -> array:
     pass
 
 def nonzero(x: array, /) -> Tuple[array, ...]: