Compare commits

..

4 Commits

Author SHA1 Message Date
Peter Zhokhov
7446e6ea34 fix typo 2020-01-31 05:23:33 -08:00
Peter Zhokhov
3bd068c82c actually close the file with the results on Monitor.close() 2020-01-31 05:09:29 -08:00
Harry Uglow
ea25b9e8b2 Monitor should close what it inherits (#1076) 2020-01-31 05:06:18 -08:00
pzhokhov
9ee399f5b2 Fix build with latest gym (#1034)
* update to use latest version of gym

* fix imports

* narrow down gym version to 0.15.4 <= gym < 0.16.0
2019-11-10 11:10:01 -08:00

View File

@@ -9,7 +9,6 @@ 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)
@@ -77,8 +76,9 @@ class Monitor(Wrapper):
self.total_steps += 1
def close(self):
if self.f is not None:
self.f.close()
super(Monitor, self).close()
if self.results_writer is not None:
self.results_writer.f.close()
def get_total_steps(self):
return self.total_steps