_pcg64.pyi 1.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041
  1. from typing import TypedDict, type_check_only
  2. from numpy._typing import _ArrayLikeInt_co
  3. from numpy.random.bit_generator import BitGenerator, SeedSequence
  4. __all__ = ["PCG64"]
  5. @type_check_only
  6. class _PCG64Internal(TypedDict):
  7. state: int
  8. inc: int
  9. @type_check_only
  10. class _PCG64State(TypedDict):
  11. bit_generator: str
  12. state: _PCG64Internal
  13. has_uint32: int
  14. uinteger: int
  15. class PCG64(BitGenerator):
  16. def __init__(self, seed: _ArrayLikeInt_co | SeedSequence | None = ...) -> None: ...
  17. def jumped(self, jumps: int = 1) -> PCG64: ...
  18. @property # type: ignore[override]
  19. def state(
  20. self,
  21. ) -> _PCG64State: ...
  22. @state.setter
  23. def state(
  24. self,
  25. value: _PCG64State,
  26. ) -> None: ...
  27. def advance(self, delta: int) -> PCG64: ...
  28. class PCG64DXSM(BitGenerator):
  29. def __init__(self, seed: _ArrayLikeInt_co | SeedSequence | None = ...) -> None: ...
  30. def jumped(self, jumps: int = 1) -> PCG64DXSM: ...
  31. @property # type: ignore[override]
  32. def state(self) -> _PCG64State: ...
  33. @state.setter
  34. def state(self, value: _PCG64State) -> None: ...
  35. def advance(self, delta: int) -> PCG64DXSM: ...