diff options
| author | Aaron Meurer <asmeurer@gmail.com> | 2021-07-09 13:57:44 -0600 |
|---|---|---|
| committer | Aaron Meurer <asmeurer@gmail.com> | 2021-07-09 13:57:44 -0600 |
| commit | 74478e2d943f4d61917d4d9122a042214eed94fd (patch) | |
| tree | 469046a04700123b9d52e184fdd884dff93a7419 /numpy/_array_api/_manipulation_functions.py | |
| parent | 29b7a69a39ac66ebd8f61c6c9c65e7e60b40b4a0 (diff) | |
| download | numpy-74478e2d943f4d61917d4d9122a042214eed94fd.tar.gz | |
Use better type signatures in the array API module
This includes returning custom dataclasses for finfo and iinfo that only
contain the properties required by the array API specification.
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 fa0c08d7b..6308bfc26 100644 --- a/numpy/_array_api/_manipulation_functions.py +++ b/numpy/_array_api/_manipulation_functions.py @@ -7,7 +7,7 @@ from typing import List, Optional, Tuple, Union import numpy as np # Note: the function name is different here -def concat(arrays: Tuple[Array, ...], /, *, axis: Optional[int] = 0) -> Array: +def concat(arrays: Union[Tuple[Array, ...], List[Array]], /, *, axis: Optional[int] = 0) -> Array: """ Array API compatible wrapper for :py:func:`np.concatenate <numpy.concatenate>`. @@ -56,7 +56,7 @@ def squeeze(x: Array, /, axis: Optional[Union[int, Tuple[int, ...]]] = None) -> """ return Array._new(np.squeeze(x._array, axis=axis)) -def stack(arrays: Tuple[Array, ...], /, *, axis: int = 0) -> Array: +def stack(arrays: Union[Tuple[Array, ...], List[Array]], /, *, axis: int = 0) -> Array: """ Array API compatible wrapper for :py:func:`np.stack <numpy.stack>`. |