wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-27 11:26:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix wrap (#6293)

Browse Source
This commit is contained in:
Eric Liang
2019-11-26 17:47:47 -08:00
committed by GitHub
parent 30b2fc1d81
commit ddc8855f41
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rllib/evaluation/rollout_worker.py
+4 -2
View File
@@ -290,13 +290,15 @@ class RolloutWorker(EvaluatorInterface):
dim=model_config.get("dim"),
framestack=model_config.get("framestack"))
if monitor_path:
env = gym.wrappers.Monitor(env, monitor_path, resume=True)
from gym import wrappers
env = wrappers.Monitor(env, monitor_path, resume=True)
return env
else:
def wrap(env):
if monitor_path:
env = gym.wrappers.Monitor(env, monitor_path, resume=True)
from gym import wrappers
env = wrappers.Monitor(env, monitor_path, resume=True)
return env
self.env = wrap(self.env)