diff options
| -rw-r--r-- | numpy/lib/ufunclike.pyi | 50 |
1 files changed, 50 insertions, 0 deletions
diff --git a/numpy/lib/ufunclike.pyi b/numpy/lib/ufunclike.pyi new file mode 100644 index 000000000..3443fa7ae --- /dev/null +++ b/numpy/lib/ufunclike.pyi @@ -0,0 +1,50 @@ +from typing import Any, overload, TypeVar, List, Union + +from numpy import floating, bool_, ndarray +from numpy.typing import ( + _ArrayLikeFloat_co, + _ArrayLikeObject_co, + _ArrayOrScalar, +) + +_ArrayType = TypeVar("_ArrayType", bound=ndarray[Any, Any]) + +__all__: List[str] + +@overload +def fix( + x: _ArrayLikeFloat_co, + out: None = ..., +) -> _ArrayOrScalar[floating[Any]]: ... +@overload +def fix( + x: _ArrayLikeObject_co, + out: None = ..., +) -> Any: ... +@overload +def fix( + x: Union[_ArrayLikeFloat_co, _ArrayLikeObject_co], + out: _ArrayType, +) -> _ArrayType: ... + +@overload +def isposinf( + x: _ArrayLikeFloat_co, + out: None = ..., +) -> _ArrayOrScalar[bool_]: ... +@overload +def isposinf( + x: _ArrayLikeFloat_co, + out: _ArrayType, +) -> _ArrayType: ... + +@overload +def isneginf( + x: _ArrayLikeFloat_co, + out: None = ..., +) -> _ArrayOrScalar[bool_]: ... +@overload +def isneginf( + x: _ArrayLikeFloat_co, + out: _ArrayType, +) -> _ArrayType: ... |