From 3bd068c82cfdd6f6ceb05f7ca7c5b8c9c8f5f380 Mon Sep 17 00:00:00 2001 From: Peter Zhokhov Date: Fri, 31 Jan 2020 05:09:29 -0800 Subject: [PATCH] actually close the file with the results on Monitor.close() --- baselines/bench/monitor.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/baselines/bench/monitor.py b/baselines/bench/monitor.py index 7a9bd48..56dbcff 100644 --- 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.close() def get_total_steps(self): return self.total_steps