def store_episode_2(self, observations, actions):
    self.observation_episode_2.append(observations)
    self.action_episode_2.append(actions)
def store_episode_2(self, observations, actions):
    self.observation_episode_2.append(observations)
    self.action_episode_2.append(actions)