TYP: Import into random module

numpy · Jul 15, 2022 · 0c21d8f · 0c21d8f
1 parent c5c79cd
commit 0c21d8f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git 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
@@ -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