diff options
Diffstat (limited to 'numpy/core/shape_base.pyi')
| -rw-r--r-- | numpy/core/shape_base.pyi | 43 |
1 files changed, 36 insertions, 7 deletions
diff --git a/numpy/core/shape_base.pyi b/numpy/core/shape_base.pyi index cea355d44..82541b55b 100644 --- a/numpy/core/shape_base.pyi +++ b/numpy/core/shape_base.pyi @@ -1,8 +1,8 @@ from collections.abc import Sequence from typing import TypeVar, overload, Any, SupportsIndex -from numpy import generic -from numpy._typing import ArrayLike, NDArray, _ArrayLike +from numpy import generic, _CastingKind +from numpy._typing import ArrayLike, NDArray, _ArrayLike, DTypeLike _SCT = TypeVar("_SCT", bound=generic) _ArrayType = TypeVar("_ArrayType", bound=NDArray[Any]) @@ -31,32 +31,61 @@ def atleast_3d(arys: ArrayLike, /) -> NDArray[Any]: ... def atleast_3d(*arys: ArrayLike) -> list[NDArray[Any]]: ... @overload -def vstack(tup: Sequence[_ArrayLike[_SCT]]) -> NDArray[_SCT]: ... +def vstack( + tup: Sequence[_ArrayLike[_SCT]], + *, + dtype: DTypeLike = ..., + casting: _CastingKind = ... +) -> NDArray[_SCT]: ... @overload -def vstack(tup: Sequence[ArrayLike]) -> NDArray[Any]: ... +def vstack( + tup: Sequence[ArrayLike], + *, + dtype: DTypeLike = ..., + casting: _CastingKind = ... +) -> NDArray[Any]: ... @overload -def hstack(tup: Sequence[_ArrayLike[_SCT]]) -> NDArray[_SCT]: ... +def hstack( + tup: Sequence[_ArrayLike[_SCT]], + *, + dtype: DTypeLike = ..., + casting: _CastingKind = ... +) -> NDArray[_SCT]: ... @overload -def hstack(tup: Sequence[ArrayLike]) -> NDArray[Any]: ... +def hstack( + tup: Sequence[ArrayLike], + *, + dtype: DTypeLike = ..., + casting: _CastingKind = ... +) -> NDArray[Any]: ... @overload def stack( arrays: Sequence[_ArrayLike[_SCT]], axis: SupportsIndex = ..., - out: None = ..., + out: None = ..., + *, + dtype: DTypeLike = ..., + casting: _CastingKind = ... ) -> NDArray[_SCT]: ... @overload def stack( arrays: Sequence[ArrayLike], axis: SupportsIndex = ..., out: None = ..., + *, + dtype: DTypeLike = ..., + casting: _CastingKind = ... ) -> NDArray[Any]: ... @overload def stack( arrays: Sequence[ArrayLike], axis: SupportsIndex = ..., out: _ArrayType = ..., + *, + dtype: DTypeLike = ..., + casting: _CastingKind = ... ) -> _ArrayType: ... @overload |