diff options
| author | Aaron Meurer <asmeurer@gmail.com> | 2021-03-30 13:53:11 -0600 |
|---|---|---|
| committer | Aaron Meurer <asmeurer@gmail.com> | 2021-03-30 13:53:11 -0600 |
| commit | 7ce435c610fcd7fee01da9d9e7ff5c1ab4ae6ef6 (patch) | |
| tree | 6cddffc39ea4219559a7789265a48c53b4192cfd /numpy/_array_api/_manipulation_functions.py | |
| parent | 479c8a24121465bbb9e0e193dc2da39cd08bdfe4 (diff) | |
| download | numpy-7ce435c610fcd7fee01da9d9e7ff5c1ab4ae6ef6.tar.gz | |
Update some annotations updated from the spec
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 4e5ca0728..033ed23a0 100644 --- a/numpy/_array_api/_manipulation_functions.py +++ b/numpy/_array_api/_manipulation_functions.py @@ -8,7 +8,7 @@ if TYPE_CHECKING: import numpy as np -def concat(arrays: Tuple[array], /, *, axis: Optional[int] = 0) -> array: +def concat(arrays: Tuple[array, ...], /, *, axis: Optional[int] = 0) -> array: """ Array API compatible wrapper for :py:func:`np.concatenate <numpy.concatenate>`. @@ -58,7 +58,7 @@ def squeeze(x: array, /, *, axis: Optional[Union[int, Tuple[int, ...]]] = None) """ return ndarray._array(np.squeeze(x._array, axis=axis)) -def stack(arrays: Tuple[array], /, *, axis: int = 0) -> array: +def stack(arrays: Tuple[array, ...], /, *, axis: int = 0) -> array: """ Array API compatible wrapper for :py:func:`np.stack <numpy.stack>`. |