Coverage for src/causalspyne/utils_random.py: 100%

8 statements  

« prev     ^ index     » next       coverage.py v7.11.0, created at 2026-05-15 16:30 +0000

1""" 

2Random number generator helpers. 

3""" 

4 

5from numbers import Integral 

6 

7from numpy.random import default_rng 

8 

9 

10def coerce_rng(rng=None, seed=None): 

11 if rng is None: 

12 return default_rng(seed) 

13 if isinstance(rng, Integral): 

14 return default_rng(rng) 

15 return rng