diff options
Diffstat (limited to 'numpy')
| -rw-r--r-- | numpy/lib/function_base.pyi | 8 | ||||
| -rw-r--r-- | numpy/typing/tests/data/reveal/lib_function_base.pyi | 7 |
2 files changed, 9 insertions, 6 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( diff --git a/numpy/typing/tests/data/reveal/lib_function_base.pyi b/numpy/typing/tests/data/reveal/lib_function_base.pyi index eebe9fbfd..259e1964f 100644 --- a/numpy/typing/tests/data/reveal/lib_function_base.pyi +++ b/numpy/typing/tests/data/reveal/lib_function_base.pyi @@ -79,9 +79,10 @@ reveal_type(np.diff("bob", n=0)) # E: str reveal_type(np.diff(AR_f8, axis=0)) # E: ndarray[Any, dtype[Any]] reveal_type(np.diff(AR_LIKE_f8, prepend=1.5)) # E: ndarray[Any, dtype[Any]] -reveal_type(np.angle(AR_f8)) # E: floating[Any] -reveal_type(np.angle(AR_c16, deg=True)) # E: complexfloating[Any, Any] -reveal_type(np.angle(AR_O)) # E: Any +reveal_type(np.angle(f8)) # E: floating[Any] +reveal_type(np.angle(AR_f8)) # E: ndarray[Any, dtype[floating[Any]]] +reveal_type(np.angle(AR_c16, deg=True)) # E: ndarray[Any, dtype[floating[Any]]] +reveal_type(np.angle(AR_O)) # E: ndarray[Any, dtype[object_]] reveal_type(np.unwrap(AR_f8)) # E: ndarray[Any, dtype[floating[Any]]] reveal_type(np.unwrap(AR_O)) # E: ndarray[Any, dtype[object_]] |