Commit 475d0d28 authored by Yuxin Wu's avatar Yuxin Wu

fix gymenv API change again

parent 879995d9
...@@ -48,8 +48,7 @@ class GymEnv(RLEnvironment): ...@@ -48,8 +48,7 @@ class GymEnv(RLEnvironment):
self._ob = self.gymenv.reset() self._ob = self.gymenv.reset()
def finish_episode(self): def finish_episode(self):
if self.use_dir is not None: self.gymenv.close()
self.gymenv.monitor.flush()
self.stats['score'].append(self.rwd_counter.sum) self.stats['score'].append(self.rwd_counter.sum)
def current_state(self): def current_state(self):
......
...@@ -42,11 +42,9 @@ class StartProcOrThread(Callback): ...@@ -42,11 +42,9 @@ class StartProcOrThread(Callback):
return return
for k in self._procs_threads: for k in self._procs_threads:
if isinstance(k, mp.Process): if isinstance(k, mp.Process):
logger.info("Stopping {} ...".format(k.name))
k.terminate() k.terminate()
k.join() k.join()
elif isinstance(k, StoppableThread): elif isinstance(k, StoppableThread):
logger.info("Stopping {} ...".format(k.name))
k.stop() k.stop()
else:
logger.warn(
"[StartProcOrThread] {} "
"is neither a Process nor a StoppableThread, won't stop it.".format(k.name))
...@@ -71,7 +71,6 @@ class FeedfreeInput(InputData): ...@@ -71,7 +71,6 @@ class FeedfreeInput(InputData):
class EnqueueThread(threading.Thread): class EnqueueThread(threading.Thread):
def __init__(self, trainer, queue, ds, input_placehdrs): def __init__(self, trainer, queue, ds, input_placehdrs):
super(EnqueueThread, self).__init__() super(EnqueueThread, self).__init__()
self.name = 'EnqueueThread' self.name = 'EnqueueThread'
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment