Update function stubs from the spec

Author: asmeurer

array_api_tests/function_stubs/creation_functions.py

@@ -12,7 +12,6 @@
 
 from ._types import (List, NestedSequence, Optional, SupportsBufferProtocol, SupportsDLPack, Tuple,
                      Union, array, device, dtype)
-from collections.abc import Sequence
 
 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
@@ -26,7 +25,7 @@ def empty(shape: Union[int, Tuple[int, ...]], *, dtype: Optional[dtype] = None,
 def empty_like(x: array, /, *, dtype: Optional[dtype] = None, device: Optional[device] = None) -> array:
     pass
 
-def eye(n_rows: int, n_cols: Optional[int] = None, /, *, k: Optional[int] = 0, dtype: Optional[dtype] = None, device: Optional[device] = None) -> array:
+def eye(n_rows: int, n_cols: Optional[int] = None, /, *, k: int = 0, dtype: Optional[dtype] = None, device: Optional[device] = None) -> array:
     pass
 
 def from_dlpack(x: object, /) -> array:
@@ -41,7 +40,7 @@ def full_like(x: array, /, fill_value: Union[int, float], *, dtype: Optional[dty
 def linspace(start: Union[int, float], stop: Union[int, float], /, num: int, *, dtype: Optional[dtype] = None, device: Optional[device] = None, endpoint: bool = True) -> array:
     pass
 
-def meshgrid(*arrays: Sequence[array], indexing: str = 'xy') -> List[array, ...]:
+def meshgrid(*arrays: array, indexing: str = 'xy') -> List[array, ...]:
     pass
 
 def ones(shape: Union[int, Tuple[int, ...]], *, dtype: Optional[dtype] = None, device: Optional[device] = None) -> array:

array_api_tests/function_stubs/data_type_functions.py

@@ -11,9 +11,8 @@
 from __future__ import annotations
 
 from ._types import List, Tuple, Union, array, dtype, finfo_object, iinfo_object
-from collections.abc import Sequence
 
-def broadcast_arrays(*arrays: Sequence[array]) -> List[array]:
+def broadcast_arrays(*arrays: array) -> List[array]:
     pass
 
 def broadcast_to(x: array, /, shape: Tuple[int, ...]) -> array:
@@ -28,7 +27,7 @@ def finfo(type: Union[dtype, array], /) -> finfo_object:
 def iinfo(type: Union[dtype, array], /) -> iinfo_object:
     pass
 
-def result_type(*arrays_and_dtypes: Sequence[Union[array, dtype]]) -> dtype:
+def result_type(*arrays_and_dtypes: Union[array, dtype]) -> dtype:
     pass
 
 __all__ = ['broadcast_arrays', 'broadcast_to', 'can_cast', 'finfo', 'iinfo', 'result_type']

array_api_tests/function_stubs/linalg.py

@@ -68,7 +68,7 @@ def solve(x1: array, x2: array, /) -> array:
 def svd(x: array, /, *, full_matrices: bool = True) -> Tuple[array, array, array]:
     pass
 
-def svdvals(x: array, /) -> Union[array, Tuple[array, ...]]:
+def svdvals(x: array, /) -> array:
     pass
 
 def tensordot(x1: array, x2: array, /, *, axes: Union[int, Tuple[Sequence[int], Sequence[int]]] = 2) -> array:
@@ -77,10 +77,10 @@ def tensordot(x1: array, x2: array, /, *, axes: Union[int, Tuple[Sequence[int],
 def trace(x: array, /, *, offset: int = 0) -> array:
     pass
 
-def vecdot(x1: array, x2: array, /, *, axis: Optional[int] = None) -> array:
+def vecdot(x1: array, x2: array, /, *, axis: int = None) -> array:
     pass
 
-def vector_norm(x: array, /, *, axis: Optional[Union[int, Tuple[int, int]]] = None, keepdims: bool = False, ord: Optional[Union[int, float, Literal[inf, -inf]]] = 2) -> array:
+def vector_norm(x: array, /, *, axis: Optional[Union[int, Tuple[int, ...]]] = None, keepdims: bool = False, ord: Union[int, float, Literal[inf, -inf]] = 2) -> array:
     pass
 
 __all__ = ['cholesky', 'cross', 'det', 'diagonal', 'eigh', 'eigvalsh', 'inv', 'matmul', 'matrix_norm', 'matrix_power', 'matrix_rank', 'matrix_transpose', 'outer', 'pinv', 'qr', 'slogdet', 'solve', 'svd', 'svdvals', 'tensordot', 'trace', 'vecdot', 'vector_norm']

array_api_tests/function_stubs/linear_algebra_functions.py

@@ -10,7 +10,7 @@
 
 from __future__ import annotations
 
-from ._types import Optional, Tuple, Union, array
+from ._types import Tuple, Union, array
 from collections.abc import Sequence
 
 def matmul(x1: array, x2: array, /) -> array:
@@ -22,7 +22,7 @@ def matrix_transpose(x: array, /) -> array:
 def tensordot(x1: array, x2: array, /, *, axes: Union[int, Tuple[Sequence[int], Sequence[int]]] = 2) -> array:
     pass
 
-def vecdot(x1: array, x2: array, /, *, axis: Optional[int] = None) -> array:
+def vecdot(x1: array, x2: array, /, *, axis: int = -1) -> array:
     pass
 
 __all__ = ['matmul', 'matrix_transpose', 'tensordot', 'vecdot']