./utils/rng.py

./utils/rng.py#

from numpy.random import default_rng

class RNG:
    def __init__(self, seed: int):
        self.rng = default_rng(seed)

    def gauss(self, mu: float, sigma: float) -> float:
        ''' Return a random float following a Gaussian distribution with mean mu and standard deviation sigma. '''
        return 0 #float(self.rng ###)

    def uniform(self, a: float, b: float) -> float:
        ''' Return a random float in the range [a, b) following a uniform distribution. '''
        return 0 #float(self.rng ###)

    def bernoulli(self, p: float) -> bool:
        ''' Return True with probability p, False with probability 1-p. '''
        return 0 #bool(self.rng ###)