Move from numpy.linalg import * to the top of the file

Otherwise the wrappers in the file do not override the NumPy definitions.

Author: asmeurer

numpy_array_api_compat/linalg.py

@@ -8,6 +8,12 @@
 import numpy as np
 from numpy.core.numeric import normalize_axis_tuple
 
+from numpy.linalg import *
+from numpy.linalg import __all__ as linalg_all
+
+# These are in the main NumPy namespace but not in numpy.linalg
+from numpy import cross, diagonal, matmul, outer, tensordot, trace
+
 class EighResult(NamedTuple):
     eigenvalues: ndarray
     eigenvectors: ndarray
@@ -104,12 +110,6 @@ def vector_norm(x: ndarray, /, *, axis: Optional[Union[int, Tuple[int, ...]]] =
 
     return res
 
-from numpy.linalg import *
-from numpy.linalg import __all__ as linalg_all
-
-# These are in the main NumPy namespace but not in numpy.linalg
-from numpy import cross, diagonal, matmul, outer, tensordot, trace
-
 __all__ = linalg_all.copy()
 __all__ += ['cross', 'diagonal', 'matmul', 'matrix_norm', 'matrix_transpose',
             'outer', 'svdvals', 'tensordot', 'trace', 'vecdot', 'vector_norm',