blob: d8895bba67cfafffbeb645d6beaa09572f7042ff (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
from typing import TypedDict, type_check_only
from numpy import uint64
from numpy._typing import _ArrayLikeInt_co
from numpy.random.bit_generator import BitGenerator, SeedSequence
from numpy.typing import NDArray
@type_check_only
class _PhiloxInternal(TypedDict):
counter: NDArray[uint64]
key: NDArray[uint64]
@type_check_only
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: _ArrayLikeInt_co | SeedSequence | None = ...,
counter: _ArrayLikeInt_co | None = ...,
key: _ArrayLikeInt_co | None = ...,
) -> 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: ...
|
