(self, arm)
When choosing a new arm, make sure we play that arm for
    tau(r+1) - tau(r) episodes.
When choosing a new arm, make sure we play that arm for
    tau(r+1) - tau(r) episodes.
When choosing a new arm make sure we play that arm for tau ( r + 1 ) - tau ( r ) episodes .
def __set_arm(self, arm):
    """
    When choosing a new arm, make sure we play that arm for
    tau(r+1) - tau(r) episodes.
    """
    self.__current_arm = arm
    self.__next_update += max(1, self.__tau(self.r[arm] + 1) - self.__tau(self.r[arm]))
    self.r[arm] += 1
def __set_arm ( self , arm ) : self . __current_arm = arm self . __next_update += max ( 1 , self . __tau ( self . r [ arm ] + 1 ) - self . __tau ( self . r [ arm ] ) ) self . r [ arm ] += 1