Use "import numpy as np" in the array_api submodule

This avoids importing everything inside the individual functions, but still is
preferred over importing the functions used explicitly, as most of them clash
with the wrapper function names.

1 files changed, 9 insertions, 14 deletions

diff --git a/numpy/_array_api/_manipulation_functions.py b/numpy/_array_api/_manipulation_functions.py
index 19e9c1cab..262c712f8 100644
--- a/numpy/_array_api/_manipulation_functions.py
+++ b/numpy/_array_api/_manipulation_functions.py
@@ -1,28 +1,23 @@
+import numpy as np
+
 def concat(arrays, /, *, axis=0):
     # Note: the function name is different here
-    from .. import concatenate
-    return concatenate(arrays, axis=axis)
+    return np.concatenate(arrays, axis=axis)
 
 def expand_dims(x, axis, /):
-    from .. import expand_dims
-    return expand_dims(x, axis)
+    return np.expand_dims(x, axis)
 
 def flip(x, /, *, axis=None):
-    from .. import flip
-    return flip(x, axis=axis)
+    return np.flip(x, axis=axis)
 
 def reshape(x, shape, /):
-    from .. import reshape
-    return reshape(x, shape)
+    return np.reshape(x, shape)
 
 def roll(x, shift, /, *, axis=None):
-    from .. import roll
-    return roll(x, shift, axis=axis)
+    return np.roll(x, shift, axis=axis)
 
 def squeeze(x, /, *, axis=None):
-    from .. import squeeze
-    return squeeze(x, axis=axis)
+    return np.squeeze(x, axis=axis)
 
 def stack(arrays, /, *, axis=0):
-    from .. import stack
-    return stack(arrays, axis=axis)
+    return np.stack(arrays, axis=axis)