moved import gif_recorder inside start_recording function to pass the…

… test.
huawei-noah · Nov 21, 2022 · b30cf41 · b30cf41
1 parent 6c3a0ef
commit b30cf41
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,gym]
+          pip install -e .[camera-obs,opendrive,rllib,test,test-notebook,torch,train]
       - 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,8 +26,6 @@
 import gym
 import gym.envs
 
-from smarts.env.wrappers.gif_recorder import GifRecorder
-
 
 class RecorderWrapper(gym.Wrapper):
     """
@@ -66,6 +64,8 @@ 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")