def choose(self):
    if (random.random() > self.eps):
        a = torch.argmax((self.r + (self.r_tol * torch.randn(self.r.shape))))
    else:
        a = random.randint(0, (self.n_arms - 1))
    return a
def choose(self):
    if (random.random() > self.eps):
        a = torch.argmax((self.r + (self.r_tol * torch.randn(self.r.shape))))
    else:
        a = random.randint(0, (self.n_arms - 1))
    return a