in support/retro_contest/__init__.py [0:0]
def reset(self, **kwargs):
if not self.start:
self.start = time.time()
else:
self.csv.writerow({
'r': self.episode_reward,
'l': self.episode_length,
't': time.time() - self.start
})
self.file.flush()
self.episode_length = 0
self.episode_reward = 0
return self.env.reset(**kwargs)