code: def on_episode_begin(self, episode, logs):
          
          self.episode_start[episode] = timeit.default_timer()
          self.observations[episode] = []
          self.rewards[episode] = []
          self.actions[episode] = []
          self.metrics[episode] = []
docs: Reset environment variables at beginning of each episode