fix typo

actually close the file with the results on Monitor.close()
2020-01-31 05:23:33 -08:00 · 2020-01-31 05:09:29 -08:00
1 changed files with 2 additions and 3 deletions
--- a/baselines/bench/monitor.py
+++ b/baselines/bench/monitor.py
@@ -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)
@@ -78,8 +77,8 @@ class Monitor(Wrapper):

    def close(self):
        super(Monitor, self).close()
-        if self.f is not None:
-            self.f.close()
+        if self.results_writer is not None:
+            self.results_writer.f.close()

    def get_total_steps(self):
        return self.total_steps
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