def add_episode(self, episode_type, lr, noise_std, buffer_size):
    if (episode_type in self.episodes.keys()):
        idx = len(self.episodes[episode_type])
        episode_name = '{}_{:03d}'.format(episode_type, idx)
        self.episodes[episode_type].append(Episode(episode_name, lr, noise_std, buffer_size, self.average_reward))
        self.current_episode = self.episodes[episode_type][(- 1)]
    else:
        raise ValueError('Invalid episode type added to result buffer')
def add_episode(self, episode_type, lr, noise_std, buffer_size):
    if (episode_type in self.episodes.keys()):
        idx = len(self.episodes[episode_type])
        episode_name = '{}_{:03d}'.format(episode_type, idx)
        self.episodes[episode_type].append(Episode(episode_name, lr, noise_std, buffer_size, self.average_reward))
        self.current_episode = self.episodes[episode_type][(- 1)]
    else:
        raise ValueError('Invalid episode type added to result buffer')