From 0c21d8f8a4ff8b0bf85460bb3e411eebbe7683c7 Mon Sep 17 00:00:00 2001 From: Kevin Sheppard Date: Fri, 15 Jul 2022 18:44:30 +0100 Subject: [PATCH] TYP: Import into random module --- numpy/random/__init__.pyi | 2 ++ numpy/typing/tests/data/reveal/random.pyi | 4 ++-- 2 files changed, 4 insertions(+), 2 deletions(-) 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