Skip to content

Commit 533d046

Browse files
committed
Fix array API squeeze() and stack()
1 parent 2199687 commit 533d046

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

numpy/_array_api/_manipulation_functions.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ def squeeze(x: array, /, *, axis: Optional[Union[int, Tuple[int, ...]]] = None)
5656
5757
See its docstring for more information.
5858
"""
59-
return ndarray._array(np.squeeze(x._array, axis=axis))
59+
return ndarray._new(np.squeeze(x._array, axis=axis))
6060

6161
def stack(arrays: Tuple[array, ...], /, *, axis: int = 0) -> array:
6262
"""
@@ -65,4 +65,4 @@ def stack(arrays: Tuple[array, ...], /, *, axis: int = 0) -> array:
6565
See its docstring for more information.
6666
"""
6767
arrays = tuple(a._array for a in arrays)
68-
return ndarray._array(np.stack(arrays, axis=axis))
68+
return ndarray._new(np.stack(arrays, axis=axis))

0 commit comments

Comments
 (0)