diff options
| author | Bas van Beek <43369155+BvB93@users.noreply.github.com> | 2022-07-12 15:50:25 +0200 |
|---|---|---|
| committer | Bas van Beek <43369155+BvB93@users.noreply.github.com> | 2022-07-12 16:14:01 +0200 |
| commit | eb24897ae5a5a844a59693804e9d809b105a566f (patch) | |
| tree | c2ce2e4e05b1b2fb16a96251bf442fa4d303fabc /numpy/lib/function_base.pyi | |
| parent | 3da5da966d4518f7312bda4872b51f9f21de50a9 (diff) | |
| download | numpy-eb24897ae5a5a844a59693804e9d809b105a566f.tar.gz | |
MAINT, TYP: Fix `np.angle` dtype-overloads
Diffstat (limited to 'numpy/lib/function_base.pyi')
| -rw-r--r-- | numpy/lib/function_base.pyi | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/numpy/lib/function_base.pyi b/numpy/lib/function_base.pyi index 6c00d26b4..c14a54c60 100644 --- a/numpy/lib/function_base.pyi +++ b/numpy/lib/function_base.pyi @@ -289,11 +289,13 @@ def interp( ) -> NDArray[complex128]: ... @overload -def angle(z: _ArrayLikeFloat_co, deg: bool = ...) -> floating[Any]: ... +def angle(z: _ComplexLike_co, deg: bool = ...) -> floating[Any]: ... @overload -def angle(z: _ArrayLikeComplex_co, deg: bool = ...) -> complexfloating[Any, Any]: ... +def angle(z: object_, deg: bool = ...) -> Any: ... @overload -def angle(z: _ArrayLikeObject_co, deg: bool = ...) -> Any: ... +def angle(z: _ArrayLikeComplex_co, deg: bool = ...) -> NDArray[floating[Any]]: ... +@overload +def angle(z: _ArrayLikeObject_co, deg: bool = ...) -> NDArray[object_]: ... @overload def unwrap( |