|
1 | | -from threading import Lock |
2 | | - |
3 | | -import numpy as np |
4 | | -import numpy.typing as npt |
5 | | -from numpy._typing import _ArrayLikeInt_co |
6 | | -from numpy.random.bit_generator import BitGenerator |
7 | | - |
8 | 1 | __all__: list[str] = [] |
9 | | - |
10 | | -# |
11 | | -def _gen_mask(max_val: np.uint64) -> np.uint64: ... |
12 | | - |
13 | | -# |
14 | | -def _rand_uint64( |
15 | | - low: int | _ArrayLikeInt_co, |
16 | | - high: int | _ArrayLikeInt_co | None = None, |
17 | | - size: int | tuple[int, ...] | None = None, |
18 | | - use_masked: bool = True, |
19 | | - closed: bool = True, |
20 | | - state: BitGenerator = ..., |
21 | | - lock: Lock = ..., |
22 | | -) -> npt.NDArray[np.uint64]: ... |
23 | | -def _rand_uint32( |
24 | | - low: int | _ArrayLikeInt_co, |
25 | | - high: int | _ArrayLikeInt_co | None = None, |
26 | | - size: int | tuple[int, ...] | None = None, |
27 | | - use_masked: bool = True, |
28 | | - closed: bool = True, |
29 | | - state: BitGenerator = ..., |
30 | | - lock: Lock = ..., |
31 | | -) -> npt.NDArray[np.uint32]: ... |
32 | | -def _rand_uint16( |
33 | | - low: int | _ArrayLikeInt_co, |
34 | | - high: int | _ArrayLikeInt_co | None = None, |
35 | | - size: int | tuple[int, ...] | None = None, |
36 | | - use_masked: bool = True, |
37 | | - closed: bool = True, |
38 | | - state: BitGenerator = ..., |
39 | | - lock: Lock = ..., |
40 | | -) -> np.uint16 | npt.NDArray[np.uint16]: ... |
41 | | -def _rand_uint8( |
42 | | - low: int | _ArrayLikeInt_co, |
43 | | - high: int | _ArrayLikeInt_co | None = None, |
44 | | - size: int | tuple[int, ...] | None = None, |
45 | | - use_masked: bool = True, |
46 | | - closed: bool = True, |
47 | | - state: BitGenerator = ..., |
48 | | - lock: Lock = ..., |
49 | | -) -> np.uint8 | npt.NDArray[np.uint8]: ... |
50 | | -def _rand_bool( |
51 | | - low: int | _ArrayLikeInt_co, |
52 | | - high: int | _ArrayLikeInt_co | None = None, |
53 | | - size: int | tuple[int, ...] | None = None, |
54 | | - use_masked: bool = True, |
55 | | - closed: bool = True, |
56 | | - state: BitGenerator = ..., |
57 | | - lock: Lock = ..., |
58 | | -) -> np.bool_ | npt.NDArray[np.bool_]: ... |
59 | | -def _rand_int64( |
60 | | - low: int | _ArrayLikeInt_co, |
61 | | - high: int | _ArrayLikeInt_co | None = None, |
62 | | - size: int | tuple[int, ...] | None = None, |
63 | | - use_masked: bool = True, |
64 | | - closed: bool = True, |
65 | | - state: BitGenerator = ..., |
66 | | - lock: Lock = ..., |
67 | | -) -> np.int64 | npt.NDArray[np.int64]: ... |
68 | | -def _rand_int32( |
69 | | - low: int | _ArrayLikeInt_co, |
70 | | - high: int | _ArrayLikeInt_co | None = None, |
71 | | - size: int | tuple[int, ...] | None = None, |
72 | | - use_masked: bool = True, |
73 | | - closed: bool = True, |
74 | | - state: BitGenerator = ..., |
75 | | - lock: Lock = ..., |
76 | | -) -> np.int32 | npt.NDArray[np.int32]: ... |
77 | | -def _rand_int16( |
78 | | - low: int | _ArrayLikeInt_co, |
79 | | - high: int | _ArrayLikeInt_co | None = None, |
80 | | - size: int | tuple[int, ...] | None = None, |
81 | | - use_masked: bool = True, |
82 | | - closed: bool = True, |
83 | | - state: BitGenerator = ..., |
84 | | - lock: Lock = ..., |
85 | | -) -> np.int16 | npt.NDArray[np.int16]: ... |
86 | | -def _rand_int8( |
87 | | - low: int | _ArrayLikeInt_co, |
88 | | - high: int | _ArrayLikeInt_co | None = None, |
89 | | - size: int | tuple[int, ...] | None = None, |
90 | | - use_masked: bool = True, |
91 | | - closed: bool = True, |
92 | | - state: BitGenerator = ..., |
93 | | - lock: Lock = ..., |
94 | | -) -> np.int8 | npt.NDArray[np.int8]: ... |
0 commit comments