diff options
| author | Bas van Beek <43369155+BvB93@users.noreply.github.com> | 2022-07-12 15:50:25 +0200 |
|---|---|---|
| committer | Charles Harris <charlesr.harris@gmail.com> | 2022-07-23 18:07:25 -0600 |
| commit | f24ca2c5165d6eb4af57fa346f5df3017b59f0dc (patch) | |
| tree | d210b57053911cac7641e47f943efc428d1911cf /numpy | |
| parent | 1cd77ce1e59ded1e8f5166a0da49cda4ddf32174 (diff) | |
| download | numpy-f24ca2c5165d6eb4af57fa346f5df3017b59f0dc.tar.gz | |
MAINT, TYP: Fix `np.angle` dtype-overloads
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_]] |