_ufunclike_impl.pyi 1.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960
  1. from typing import Any, TypeVar, overload
  2. from typing_extensions import deprecated
  3. import numpy as np
  4. from numpy import floating, object_
  5. from numpy._typing import (
  6. NDArray,
  7. _ArrayLikeFloat_co,
  8. _ArrayLikeObject_co,
  9. _FloatLike_co,
  10. )
  11. __all__ = ["fix", "isneginf", "isposinf"]
  12. _ArrayT = TypeVar("_ArrayT", bound=NDArray[Any])
  13. @overload
  14. @deprecated("np.fix will be deprecated in NumPy 2.5 in favor of np.trunc", category=PendingDeprecationWarning)
  15. def fix(x: _FloatLike_co, out: None = None) -> floating: ...
  16. @overload
  17. @deprecated("np.fix will be deprecated in NumPy 2.5 in favor of np.trunc", category=PendingDeprecationWarning)
  18. def fix(x: _ArrayLikeFloat_co, out: None = None) -> NDArray[floating]: ...
  19. @overload
  20. @deprecated("np.fix will be deprecated in NumPy 2.5 in favor of np.trunc", category=PendingDeprecationWarning)
  21. def fix(x: _ArrayLikeObject_co, out: None = None) -> NDArray[object_]: ...
  22. @overload
  23. @deprecated("np.fix will be deprecated in NumPy 2.5 in favor of np.trunc", category=PendingDeprecationWarning)
  24. def fix(x: _ArrayLikeFloat_co | _ArrayLikeObject_co, out: _ArrayT) -> _ArrayT: ...
  25. @overload
  26. def isposinf( # type: ignore[misc]
  27. x: _FloatLike_co,
  28. out: None = None,
  29. ) -> np.bool: ...
  30. @overload
  31. def isposinf(
  32. x: _ArrayLikeFloat_co,
  33. out: None = None,
  34. ) -> NDArray[np.bool]: ...
  35. @overload
  36. def isposinf(
  37. x: _ArrayLikeFloat_co,
  38. out: _ArrayT,
  39. ) -> _ArrayT: ...
  40. @overload
  41. def isneginf( # type: ignore[misc]
  42. x: _FloatLike_co,
  43. out: None = None,
  44. ) -> np.bool: ...
  45. @overload
  46. def isneginf(
  47. x: _ArrayLikeFloat_co,
  48. out: None = None,
  49. ) -> NDArray[np.bool]: ...
  50. @overload
  51. def isneginf(
  52. x: _ArrayLikeFloat_co,
  53. out: _ArrayT,
  54. ) -> _ArrayT: ...