Compare commits

..

3 Commits

Author SHA1 Message Date
Peter Zhokhov
8d9e20fec3 narrow down gym version to 0.15.4 <= gym < 0.16.0 2019-11-10 11:08:59 -08:00
Peter Zhokhov
fc23c78c77 fix imports 2019-11-08 15:39:57 -08:00
Peter Zhokhov
25f750d84f update to use latest version of gym 2019-11-08 15:31:40 -08:00

View File

@@ -9,6 +9,7 @@ import json
class Monitor(Wrapper):
EXT = "monitor.csv"
f = None
def __init__(self, env, filename, allow_early_resets=False, reset_keywords=(), info_keywords=()):
Wrapper.__init__(self, env=env)
@@ -76,9 +77,8 @@ class Monitor(Wrapper):
self.total_steps += 1
def close(self):
super(Monitor, self).close()
if self.results_writer is not None:
self.results_writer.f.close()
if self.f is not None:
self.f.close()
def get_total_steps(self):
return self.total_steps