Coverage for bilby/core/utils/random.py: 100%

1from numpy.random import default_rng, SeedSequence 

2 

3 

4def __getattr__(name): 

5 if name == "rng": 

6 return Generator.rng 

7 

8 

9class Generator: 

10 rng = default_rng() 

11 

12 

13def seed(seed): 

14 Generator.rng = default_rng(seed) 

15 

16 

17def generate_seeds(nseeds): 

18 return SeedSequence(Generator.rng.integers(0, 2**63 - 1, size=4)).spawn(nseeds)