_asarray.pyi 1.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677
  1. import sys
  2. from typing import TypeVar, Union, Iterable, overload
  3. from numpy import ndarray, _OrderKACF
  4. from numpy.typing import ArrayLike, DTypeLike
  5. if sys.version_info >= (3, 8):
  6. from typing import Literal
  7. else:
  8. from typing_extensions import Literal
  9. _ArrayType = TypeVar("_ArrayType", bound=ndarray)
  10. def asarray(
  11. a: object,
  12. dtype: DTypeLike = ...,
  13. order: _OrderKACF = ...,
  14. *,
  15. like: ArrayLike = ...
  16. ) -> ndarray: ...
  17. @overload
  18. def asanyarray(
  19. a: _ArrayType,
  20. dtype: None = ...,
  21. order: _OrderKACF = ...,
  22. *,
  23. like: ArrayLike = ...
  24. ) -> _ArrayType: ...
  25. @overload
  26. def asanyarray(
  27. a: object,
  28. dtype: DTypeLike = ...,
  29. order: _OrderKACF = ...,
  30. *,
  31. like: ArrayLike = ...
  32. ) -> ndarray: ...
  33. def ascontiguousarray(
  34. a: object, dtype: DTypeLike = ..., *, like: ArrayLike = ...
  35. ) -> ndarray: ...
  36. def asfortranarray(
  37. a: object, dtype: DTypeLike = ..., *, like: ArrayLike = ...
  38. ) -> ndarray: ...
  39. _Requirements = Literal[
  40. "C", "C_CONTIGUOUS", "CONTIGUOUS",
  41. "F", "F_CONTIGUOUS", "FORTRAN",
  42. "A", "ALIGNED",
  43. "W", "WRITEABLE",
  44. "O", "OWNDATA"
  45. ]
  46. _E = Literal["E", "ENSUREARRAY"]
  47. _RequirementsWithE = Union[_Requirements, _E]
  48. @overload
  49. def require(
  50. a: _ArrayType,
  51. dtype: None = ...,
  52. requirements: Union[None, _Requirements, Iterable[_Requirements]] = ...,
  53. *,
  54. like: ArrayLike = ...
  55. ) -> _ArrayType: ...
  56. @overload
  57. def require(
  58. a: object,
  59. dtype: DTypeLike = ...,
  60. requirements: Union[_E, Iterable[_RequirementsWithE]] = ...,
  61. *,
  62. like: ArrayLike = ...
  63. ) -> ndarray: ...
  64. @overload
  65. def require(
  66. a: object,
  67. dtype: DTypeLike = ...,
  68. requirements: Union[None, _Requirements, Iterable[_Requirements]] = ...,
  69. *,
  70. like: ArrayLike = ...
  71. ) -> ndarray: ...