def __init__(self, total_episodes_in_epoch: Optional[int]=None):
    super().__init__()
    self.total_episodes_in_epoch = total_episodes_in_epoch
def __init__(self, total_episodes_in_epoch: Optional[int]=None):
    super().__init__()
    self.total_episodes_in_epoch = total_episodes_in_epoch