def on_step_end(self, step, logs):
episode = logs['<STR_LIT>']<EOL>self.observations[episode].append(logs['<STR_LIT>'])<EOL>self.rewards[episode].append(logs['<STR_LIT>'])<EOL>self.actions[episode].append(logs['<STR_LIT:action>'])<EOL>self.metrics[episode].append(logs['<STR_LIT>'])<EOL>self.step += <NUM_LIT:1><EOL>
Update statistics of episode after each step