Runner requires manual call to close() so calling-code can work with …

…"dirty" runner
lefnire · Jan 13, 2018 · bda3e66 · bda3e66
1 parent 07344b8
commit bda3e66
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tensorforce/execution/runner.py b/tensorforce/execution/runner.py
@@ -64,6 +64,10 @@ def reset(self, history=None):
         self.episode_timesteps = history.get('episode_timesteps', list())
         self.episode_times = history.get('episode_times', list())
 
+    def close(self):
+        self.agent.close()
+        self.environment.close()
+
     def run(
         self,
         timesteps=None,
@@ -147,6 +151,3 @@ def run(
                     self.agent.should_stop():
                 # agent.episode / agent.timestep are globally updated
                 break
-
-        self.agent.close()
-        self.environment.close()