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

fix gymenv API change again

parent 879995d9
......@@ -48,8 +48,7 @@ class GymEnv(RLEnvironment):
self._ob = self.gymenv.reset()
def finish_episode(self):
if self.use_dir is not None:
self.gymenv.monitor.flush()
self.gymenv.close()
self.stats['score'].append(self.rwd_counter.sum)
def current_state(self):
......
......@@ -42,11 +42,9 @@ class StartProcOrThread(Callback):
return
for k in self._procs_threads:
if isinstance(k, mp.Process):
logger.info("Stopping {} ...".format(k.name))
k.terminate()
k.join()
elif isinstance(k, StoppableThread):
logger.info("Stopping {} ...".format(k.name))
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):
class EnqueueThread(threading.Thread):
def __init__(self, trainer, queue, ds, input_placehdrs):
super(EnqueueThread, self).__init__()
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