diff options
Diffstat (limited to 'numpy/lib')
| -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( |