diff options
| author | Aaron Meurer <asmeurer@gmail.com> | 2021-05-10 15:18:42 -0600 |
|---|---|---|
| committer | Aaron Meurer <asmeurer@gmail.com> | 2021-05-10 15:18:42 -0600 |
| commit | 533d0468f12e89b1b4b299f0344a31378853b012 (patch) | |
| tree | 0cba4c4fab4b1f2d366412d9f79a68af08b049a1 /numpy/_array_api/_manipulation_functions.py | |
| parent | 219968727b6a28e8564a22284cb630a808bc0c04 (diff) | |
| download | numpy-533d0468f12e89b1b4b299f0344a31378853b012.tar.gz | |
Fix array API squeeze() and stack()
Diffstat (limited to 'numpy/_array_api/_manipulation_functions.py')
| -rw-r--r-- | numpy/_array_api/_manipulation_functions.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/numpy/_array_api/_manipulation_functions.py b/numpy/_array_api/_manipulation_functions.py index fb9e25baa..b461f6b6b 100644 --- a/numpy/_array_api/_manipulation_functions.py +++ b/numpy/_array_api/_manipulation_functions.py @@ -56,7 +56,7 @@ def squeeze(x: array, /, *, axis: Optional[Union[int, Tuple[int, ...]]] = None) See its docstring for more information. """ - return ndarray._array(np.squeeze(x._array, axis=axis)) + return ndarray._new(np.squeeze(x._array, axis=axis)) def stack(arrays: Tuple[array, ...], /, *, axis: int = 0) -> array: """ @@ -65,4 +65,4 @@ def stack(arrays: Tuple[array, ...], /, *, axis: int = 0) -> array: See its docstring for more information. """ arrays = tuple(a._array for a in arrays) - return ndarray._array(np.stack(arrays, axis=axis)) + return ndarray._new(np.stack(arrays, axis=axis)) |