summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorKevin Sheppard <kevin.k.sheppard@gmail.com>2022-07-15 18:44:30 +0100
committerKevin Sheppard <kevin.k.sheppard@gmail.com>2022-08-11 15:28:25 +0100
commitcf3ca1587bcadfb0747b7ec31c3e0fdc988ef1e8 (patch)
treea8984edc7ba35a447877b8165deb583b15628324
parent4b8cd57bedd1b10fd4cebb2eaaff2acead71b70c (diff)
downloadnumpy-cf3ca1587bcadfb0747b7ec31c3e0fdc988ef1e8.tar.gz
TYP: Import into random module
-rw-r--r--numpy/random/__init__.pyi2
-rw-r--r--numpy/typing/tests/data/reveal/random.pyi4
2 files changed, 4 insertions, 2 deletions
diff --git a/numpy/random/__init__.pyi b/numpy/random/__init__.pyi
index 32bd64a0b..99ef6f3e2 100644
--- a/numpy/random/__init__.pyi
+++ b/numpy/random/__init__.pyi
@@ -23,6 +23,7 @@ from numpy.random.mtrand import (
f as f,
gamma as gamma,
geometric as geometric,
+ get_bit_generator as get_bit_generator,
get_state as get_state,
gumbel as gumbel,
hypergeometric as hypergeometric,
@@ -50,6 +51,7 @@ from numpy.random.mtrand import (
rayleigh as rayleigh,
sample as sample,
seed as seed,
+ set_bit_generator as set_bit_generator,
set_state as set_state,
shuffle as shuffle,
standard_cauchy as standard_cauchy,
diff --git a/numpy/typing/tests/data/reveal/random.pyi b/numpy/typing/tests/data/reveal/random.pyi
index b65e47006..67a5d3e7a 100644
--- a/numpy/typing/tests/data/reveal/random.pyi
+++ b/numpy/typing/tests/data/reveal/random.pyi
@@ -1538,5 +1538,5 @@ reveal_type(random_st.tomaxint()) # E: int
reveal_type(random_st.tomaxint(1)) # E: ndarray[Any, dtype[{int_}]]
reveal_type(random_st.tomaxint((1,))) # E: ndarray[Any, dtype[{int_}]]
-reveal_type(np.random.set_bit_generator(pcg64)) # E: BitGenerator
-reveal_type(np.random.get_bit_generator()) # E: None
+reveal_type(np.random.set_bit_generator(pcg64)) # E: None
+reveal_type(np.random.get_bit_generator()) # E: BitGenerator