More cleanup

Author: adonath

array_api_compat/common/_linalg.py

@@ -11,7 +11,7 @@
 else:
  from numpy.core.numeric import normalize_axis_tuple
 
-from ._aliases import matmul, matrix_transpose, tensordot, vecdot, isdtype
+from ._aliases import matrix_transpose, isdtype
 from .._internal import get_xp
 
 # These are in the main NumPy namespace but not in numpy.linalg
@@ -149,10 +149,4 @@ def trace(x: ndarray, /, xp, *, offset: int = 0, dtype=None, **kwargs) -> ndarra
  dtype = xp.float64
  elif x.dtype == xp.complex64:
  dtype = xp.complex128
- return xp.asarray(xp.trace(x, offset=offset, dtype=dtype, axis1=-2, axis2=-1, **kwargs))
-
-__all__ = ['cross', 'matmul', 'outer', 'tensordot', 'EighResult',
- 'QRResult', 'SlogdetResult', 'SVDResult', 'eigh', 'qr', 'slogdet',
- 'svd', 'cholesky', 'matrix_rank', 'pinv', 'matrix_norm',
- 'matrix_transpose', 'svdvals', 'vecdot', 'vector_norm', 'diagonal',
- 'trace']
+ return xp.asarray(xp.trace(x, offset=offset, dtype=dtype, axis1=-2, axis2=-1, **kwargs))

array_api_compat/common/_typing.py

@@ -18,3 +18,6 @@ def __getitem__(self, key: int, /) -> _T_co | NestedSequence[_T_co]: ...
  def __len__(self, /) -> int: ...
 
 SupportsBufferProtocol = Any
+
+Array = Any
+Device = Any