1  from typing import Any, TypedDict
 2  
 3  from numpy import dtype, ndarray, uint64
 4  from numpy.random.bit_generator import BitGenerator, SeedSequence
 5  from numpy._typing import _ArrayLikeInt_co
 6  
 7  class _PhiloxInternal(TypedDict):
 8      counter: ndarray[Any, dtype[uint64]]
 9      key: ndarray[Any, dtype[uint64]]
10  
11  class _PhiloxState(TypedDict):
12      bit_generator: str
13      state: _PhiloxInternal
14      buffer: ndarray[Any, dtype[uint64]]
15      buffer_pos: int
16      has_uint32: int
17      uinteger: int
18  
19  class Philox(BitGenerator):
20      def __init__(
21          self,
22          seed: None | _ArrayLikeInt_co | SeedSequence = ...,
23          counter: None | _ArrayLikeInt_co = ...,
24          key: None | _ArrayLikeInt_co = ...,
25      ) -> None: ...
26      @property
27      def state(
28          self,
29      ) -> _PhiloxState: ...
30      @state.setter
31      def state(
32          self,
33          value: _PhiloxState,
34      ) -> None: ...
35      def jumped(self, jumps: int = ...) -> Philox: ...
36      def advance(self, delta: int) -> Philox: ...