|
from typing import TypedDict |
|
|
|
from numpy import uint64 |
|
from numpy.typing import NDArray |
|
from numpy.random.bit_generator import BitGenerator, SeedSequence |
|
from numpy._typing import _ArrayLikeInt_co |
|
|
|
class _PhiloxInternal(TypedDict): |
|
counter: NDArray[uint64] |
|
key: NDArray[uint64] |
|
|
|
class _PhiloxState(TypedDict): |
|
bit_generator: str |
|
state: _PhiloxInternal |
|
buffer: NDArray[uint64] |
|
buffer_pos: int |
|
has_uint32: int |
|
uinteger: int |
|
|
|
class Philox(BitGenerator): |
|
def __init__( |
|
self, |
|
seed: None | _ArrayLikeInt_co | SeedSequence = ..., |
|
counter: None | _ArrayLikeInt_co = ..., |
|
key: None | _ArrayLikeInt_co = ..., |
|
) -> None: ... |
|
@property |
|
def state( |
|
self, |
|
) -> _PhiloxState: ... |
|
@state.setter |
|
def state( |
|
self, |
|
value: _PhiloxState, |
|
) -> None: ... |
|
def jumped(self, jumps: int = ...) -> Philox: ... |
|
def advance(self, delta: int) -> Philox: ... |
|
|
