def __init__(self, n_arms, lamb=1):
    super(mab_user, self).__init__()
    self.t = torch.tensor(1.0)
    self.r = torch.zeros(n_arms)
    self.n = torch.zeros(n_arms)
    self.id = (- 1)
    self.returns = 0
    self.lamb = lamb
def __init__(self, n_arms, lamb=1):
    super(mab_user, self).__init__()
    self.t = torch.tensor(1.0)
    self.r = torch.zeros(n_arms)
    self.n = torch.zeros(n_arms)
    self.id = (- 1)
    self.returns = 0
    self.lamb = lamb