Compare commits
3 Commits
fix_monito
...
fix_build
Author | SHA1 | Date | |
---|---|---|---|
|
8d9e20fec3 | ||
|
fc23c78c77 | ||
|
25f750d84f |
@@ -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
|
||||
|
Reference in New Issue
Block a user