From 741cc6c411645034c8aaed75f4db16b53ff663f4 Mon Sep 17 00:00:00 2001 From: AisenGinn Date: Mon, 21 Nov 2022 16:11:18 -0500 Subject: [PATCH] revert changing. --- .github/workflows/ci-base-tests-linux.yml | 2 +- smarts/env/wrappers/recorder_wrapper.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/ci-base-tests-linux.yml b/.github/workflows/ci-base-tests-linux.yml index f5b831a899..29d01f53f3 100644 --- a/.github/workflows/ci-base-tests-linux.yml +++ b/.github/workflows/ci-base-tests-linux.yml @@ -39,7 +39,7 @@ jobs: . ${{env.venv_dir}}/bin/activate pip install --upgrade pip pip install --upgrade wheel - pip install -e .[camera-obs,opendrive,rllib,test,test-notebook,torch,train] + pip install -e .[camera-obs,opendrive,rllib,test,test-notebook,torch,train,gym] - name: Run smoke tests run: | . ${{env.venv_dir}}/bin/activate diff --git a/smarts/env/wrappers/recorder_wrapper.py b/smarts/env/wrappers/recorder_wrapper.py index fbb44f43e8..10b4eab5c1 100644 --- a/smarts/env/wrappers/recorder_wrapper.py +++ b/smarts/env/wrappers/recorder_wrapper.py @@ -26,6 +26,8 @@ import gym import gym.envs +from smarts.env.wrappers.gif_recorder import GifRecorder + class RecorderWrapper(gym.Wrapper): """ @@ -64,8 +66,6 @@ def start_recording(self): """ Start the gif recorder and capture the first frame. """ - from smarts.env.wrappers.gif_recorder import GifRecorder - if self.gif_recorder is None: self.gif_recorder = GifRecorder(self.video_name_folder, self.env) image = super().render(mode="rgb_array")