class BernoulliArm(object):

    def __init__(self, mu):
        self.mu = mu

    def draw(self):
        return np.random.binomial(n=1, p=self.mu)