revert changing.

huawei-noah · Nov 21, 2022 · 741cc6c · 741cc6c
1 parent b30cf41
commit 741cc6c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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")