def __init__(self, arms):
    self.no_of_arms = arms
    self.arms = [np.random.normal(0, 1) for _ in range(arms)]
def __init__(self, arms):
    self.no_of_arms = arms
    self.arms = [np.random.normal(0, 1) for _ in range(arms)]