From ddc8855f41675520b869f6bb045793a309330e30 Mon Sep 17 00:00:00 2001 From: Eric Liang Date: Tue, 26 Nov 2019 17:47:47 -0800 Subject: [PATCH] Fix wrap (#6293) --- rllib/evaluation/rollout_worker.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/rllib/evaluation/rollout_worker.py b/rllib/evaluation/rollout_worker.py index aa04ae274..ac9edd344 100644 --- a/rllib/evaluation/rollout_worker.py +++ b/rllib/evaluation/rollout_worker.py @@ -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)