def reset_episode_vars(self):
    self.noise.reset()
    state = self.task.reset()
    self.last_state = state
    self.reward_total = 0
    self.score = 0
    self.count = 0
    self.max_z = 0
    return state
def reset_episode_vars(self):
    self.noise.reset()
    state = self.task.reset()
    self.last_state = state
    self.reward_total = 0
    self.score = 0
    self.count = 0
    self.max_z = 0
    return state