Fix #631

Farama-Foundation · Nov 12, 2024 · 116c23f · 116c23f
1 parent def6f4f
commit 116c23f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/highway_env/envs/common/abstract.py b/highway_env/envs/common/abstract.py
@@ -325,6 +325,7 @@ def get_available_actions(self) -> list[int]:
     def set_record_video_wrapper(self, wrapper: RecordVideo):
         self._record_video_wrapper = wrapper
         self.update_metadata()
+        self._record_video_wrapper.frames_per_sec = self.metadata["render_fps"]
 
     def _automatic_rendering(self) -> None:
         """
@@ -334,8 +335,8 @@ def _automatic_rendering(self) -> None:
         If a RecordVideo wrapper has been set, use it to capture intermediate frames.
         """
         if self.viewer is not None and self.enable_auto_render:
-            if self._record_video_wrapper and self._record_video_wrapper.video_recorder:
-                self._record_video_wrapper.video_recorder.capture_frame()
+            if self._record_video_wrapper:
+                self._record_video_wrapper._capture_frame()
             else:
                 self.render()
 

diff --git a/scripts/sb3_highway_dqn.py b/scripts/sb3_highway_dqn.py
@@ -40,8 +40,8 @@
     env = RecordVideo(
         env, video_folder="highway_dqn/videos", episode_trigger=lambda e: True
     )
-    env.unwrapped.set_record_video_wrapper(env)
     env.unwrapped.config["simulation_frequency"] = 15  # Higher FPS for rendering
+    env.unwrapped.set_record_video_wrapper(env)
 
     for videos in range(10):
         done = truncated = False