Update the latest version of the spec

Some linear algebra function stubs are removed because they are now in
extensions, and we need to update the generate_stubs script to be able to
parse those.

Author: asmeurer

array_api_tests/function_stubs/__init__.py

@@ -29,9 +29,9 @@
 
 __all__ += ['abs', 'acos', 'acosh', 'add', 'asin', 'asinh', 'atan', 'atan2', 'atanh', 'bitwise_and', 'bitwise_left_shift', 'bitwise_invert', 'bitwise_or', 'bitwise_right_shift', 'bitwise_xor', 'ceil', 'cos', 'cosh', 'divide', 'equal', 'exp', 'expm1', 'floor', 'floor_divide', 'greater', 'greater_equal', 'isfinite', 'isinf', 'isnan', 'less', 'less_equal', 'log', 'log1p', 'log2', 'log10', 'logaddexp', 'logical_and', 'logical_not', 'logical_or', 'logical_xor', 'multiply', 'negative', 'not_equal', 'positive', 'pow', 'remainder', 'round', 'sign', 'sin', 'sinh', 'square', 'sqrt', 'subtract', 'tan', 'tanh', 'trunc']
 
-from .linear_algebra_functions import cholesky, cross, det, diagonal, dot, eig, eigvalsh, einsum, inv, lstsq, matmul, matrix_power, matrix_rank, norm, outer, pinv, qr, slogdet, solve, svd, trace, transpose
+from .linear_algebra_functions import einsum, matmul, tensordot, transpose, vecdot
 
-__all__ += ['cholesky', 'cross', 'det', 'diagonal', 'dot', 'eig', 'eigvalsh', 'einsum', 'inv', 'lstsq', 'matmul', 'matrix_power', 'matrix_rank', 'norm', 'outer', 'pinv', 'qr', 'slogdet', 'solve', 'svd', 'trace', 'transpose']
+__all__ += ['einsum', 'matmul', 'tensordot', 'transpose', 'vecdot']
 
 from .manipulation_functions import concat, expand_dims, flip, reshape, roll, squeeze, stack
 

array_api_tests/function_stubs/array_object.py

@@ -19,13 +19,13 @@ def __abs__(self: array, /) -> array:
     """
     pass
 
-def __add__(self: array, other: array, /) -> array:
+def __add__(self: array, other: Union[int, float, array], /) -> array:
     """
     Note: __add__ is a method of the array object.
     """
     pass
 
-def __and__(self: array, other: array, /) -> array:
+def __and__(self: array, other: Union[int, bool, array], /) -> array:
     """
     Note: __and__ is a method of the array object.
     """
@@ -55,7 +55,7 @@ def __dlpack_device__(self: array, /) -> Tuple[IntEnum, int]:
     """
     pass
 
-def __eq__(self: array, other: array, /) -> array:
+def __eq__(self: array, other: Union[int, float, bool, array], /) -> array:
     """
     Note: __eq__ is a method of the array object.
     """
@@ -67,13 +67,13 @@ def __float__(self: array, /) -> float:
     """
     pass
 
-def __floordiv__(self: array, other: array, /) -> array:
+def __floordiv__(self: array, other: Union[int, float, array], /) -> array:
     """
     Note: __floordiv__ is a method of the array object.
     """
     pass
 
-def __ge__(self: array, other: array, /) -> array:
+def __ge__(self: array, other: Union[int, float, array], /) -> array:
     """
     Note: __ge__ is a method of the array object.
     """
@@ -85,7 +85,7 @@ def __getitem__(self: array, key: Union[int, slice, ellipsis, Tuple[Union[int, s
     """
     pass
 
-def __gt__(self: array, other: array, /) -> array:
+def __gt__(self: array, other: Union[int, float, array], /) -> array:
     """
     Note: __gt__ is a method of the array object.
     """
@@ -103,7 +103,7 @@ def __invert__(self: array, /) -> array:
     """
     pass
 
-def __le__(self: array, other: array, /) -> array:
+def __le__(self: array, other: Union[int, float, array], /) -> array:
     """
     Note: __le__ is a method of the array object.
     """
@@ -115,13 +115,13 @@ def __len__(self, /):
     """
     pass
 
-def __lshift__(self: array, other: array, /) -> array:
+def __lshift__(self: array, other: Union[int, array], /) -> array:
     """
     Note: __lshift__ is a method of the array object.
     """
     pass
 
-def __lt__(self: array, other: array, /) -> array:
+def __lt__(self: array, other: Union[int, float, array], /) -> array:
     """
     Note: __lt__ is a method of the array object.
     """
@@ -133,19 +133,19 @@ def __matmul__(self: array, other: array, /) -> array:
     """
     pass
 
-def __mod__(self: array, other: array, /) -> array:
+def __mod__(self: array, other: Union[int, float, array], /) -> array:
     """
     Note: __mod__ is a method of the array object.
     """
     pass
 
-def __mul__(self: array, other: array, /) -> array:
+def __mul__(self: array, other: Union[int, float, array], /) -> array:
     """
     Note: __mul__ is a method of the array object.
     """
     pass
 
-def __ne__(self: array, other: array, /) -> array:
+def __ne__(self: array, other: Union[int, float, bool, array], /) -> array:
     """
     Note: __ne__ is a method of the array object.
     """
@@ -157,7 +157,7 @@ def __neg__(self: array, /) -> array:
     """
     pass
 
-def __or__(self: array, other: array, /) -> array:
+def __or__(self: array, other: Union[int, bool, array], /) -> array:
     """
     Note: __or__ is a method of the array object.
     """
@@ -169,13 +169,13 @@ def __pos__(self: array, /) -> array:
     """
     pass
 
-def __pow__(self: array, other: array, /) -> array:
+def __pow__(self: array, other: Union[int, float, array], /) -> array:
     """
     Note: __pow__ is a method of the array object.
     """
     pass
 
-def __rshift__(self: array, other: array, /) -> array:
+def __rshift__(self: array, other: Union[int, array], /) -> array:
     """
     Note: __rshift__ is a method of the array object.
     """
@@ -187,67 +187,67 @@ def __setitem__(self, key, value, /):
     """
     pass
 
-def __sub__(self: array, other: array, /) -> array:
+def __sub__(self: array, other: Union[int, float, array], /) -> array:
     """
     Note: __sub__ is a method of the array object.
     """
     pass
 
-def __truediv__(self: array, other: array, /) -> array:
+def __truediv__(self: array, other: Union[int, float, array], /) -> array:
     """
     Note: __truediv__ is a method of the array object.
     """
     pass
 
-def __xor__(self: array, other: array, /) -> array:
+def __xor__(self: array, other: Union[int, bool, array], /) -> array:
     """
     Note: __xor__ is a method of the array object.
     """
     pass
 
-def __iadd__(self: array, other: array, /) -> array:
+def __iadd__(self: array, other: Union[int, float, array], /) -> array:
     """
     Note: __iadd__ is a method of the array object.
     """
     pass
 
-def __radd__(self: array, other: array, /) -> array:
+def __radd__(self: array, other: Union[int, float, array], /) -> array:
     """
     Note: __radd__ is a method of the array object.
     """
     pass
 
-def __iand__(self: array, other: array, /) -> array:
+def __iand__(self: array, other: Union[int, bool, array], /) -> array:
     """
     Note: __iand__ is a method of the array object.
     """
     pass
 
-def __rand__(self: array, other: array, /) -> array:
+def __rand__(self: array, other: Union[int, bool, array], /) -> array:
     """
     Note: __rand__ is a method of the array object.
     """
     pass
 
-def __ifloordiv__(self: array, other: array, /) -> array:
+def __ifloordiv__(self: array, other: Union[int, float, array], /) -> array:
     """
     Note: __ifloordiv__ is a method of the array object.
     """
     pass
 
-def __rfloordiv__(self: array, other: array, /) -> array:
+def __rfloordiv__(self: array, other: Union[int, float, array], /) -> array:
     """
     Note: __rfloordiv__ is a method of the array object.
     """
     pass
 
-def __ilshift__(self: array, other: array, /) -> array:
+def __ilshift__(self: array, other: Union[int, array], /) -> array:
     """
     Note: __ilshift__ is a method of the array object.
     """
     pass
 
-def __rlshift__(self: array, other: array, /) -> array:
+def __rlshift__(self: array, other: Union[int, array], /) -> array:
     """
     Note: __rlshift__ is a method of the array object.
     """
@@ -265,97 +265,97 @@ def __rmatmul__(self: array, other: array, /) -> array:
     """
     pass
 
-def __imod__(self: array, other: array, /) -> array:
+def __imod__(self: array, other: Union[int, float, array], /) -> array:
     """
     Note: __imod__ is a method of the array object.
     """
     pass
 
-def __rmod__(self: array, other: array, /) -> array:
+def __rmod__(self: array, other: Union[int, float, array], /) -> array:
     """
     Note: __rmod__ is a method of the array object.
     """
     pass
 
-def __imul__(self: array, other: array, /) -> array:
+def __imul__(self: array, other: Union[int, float, array], /) -> array:
     """
     Note: __imul__ is a method of the array object.
     """
     pass
 
-def __rmul__(self: array, other: array, /) -> array:
+def __rmul__(self: array, other: Union[int, float, array], /) -> array:
     """
     Note: __rmul__ is a method of the array object.
     """
     pass
 
-def __ior__(self: array, other: array, /) -> array:
+def __ior__(self: array, other: Union[int, bool, array], /) -> array:
     """
     Note: __ior__ is a method of the array object.
     """
     pass
 
-def __ror__(self: array, other: array, /) -> array:
+def __ror__(self: array, other: Union[int, bool, array], /) -> array:
     """
     Note: __ror__ is a method of the array object.
     """
     pass
 
-def __ipow__(self: array, other: array, /) -> array:
+def __ipow__(self: array, other: Union[int, float, array], /) -> array:
     """
     Note: __ipow__ is a method of the array object.
     """
     pass
 
-def __rpow__(self: array, other: array, /) -> array:
+def __rpow__(self: array, other: Union[int, float, array], /) -> array:
     """
     Note: __rpow__ is a method of the array object.
     """
     pass
 
-def __irshift__(self: array, other: array, /) -> array:
+def __irshift__(self: array, other: Union[int, array], /) -> array:
     """
     Note: __irshift__ is a method of the array object.
     """
     pass
 
-def __rrshift__(self: array, other: array, /) -> array:
+def __rrshift__(self: array, other: Union[int, array], /) -> array:
     """
     Note: __rrshift__ is a method of the array object.
     """
     pass
 
-def __isub__(self: array, other: array, /) -> array:
+def __isub__(self: array, other: Union[int, float, array], /) -> array:
     """
     Note: __isub__ is a method of the array object.
     """
     pass
 
-def __rsub__(self: array, other: array, /) -> array:
+def __rsub__(self: array, other: Union[int, float, array], /) -> array:
     """
     Note: __rsub__ is a method of the array object.
     """
     pass
 
-def __itruediv__(self: array, other: array, /) -> array:
+def __itruediv__(self: array, other: Union[int, float, array], /) -> array:
     """
     Note: __itruediv__ is a method of the array object.
     """
     pass
 
-def __rtruediv__(self: array, other: array, /) -> array:
+def __rtruediv__(self: array, other: Union[int, float, array], /) -> array:
     """
     Note: __rtruediv__ is a method of the array object.
     """
     pass
 
-def __ixor__(self: array, other: array, /) -> array:
+def __ixor__(self: array, other: Union[int, bool, array], /) -> array:
     """
     Note: __ixor__ is a method of the array object.
     """
     pass
 
-def __rxor__(self: array, other: array, /) -> array:
+def __rxor__(self: array, other: Union[int, bool, array], /) -> array:
     """
     Note: __rxor__ is a method of the array object.
     """