diff --git a/numpy/random/__init__.pyi b/numpy/random/__init__.pyi index 32bd64a0b01d..99ef6f3e2f2a 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 b65e47006408..67a5d3e7aad8 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