quick fix

vwxyzjn · Nov 28, 2023 · f98f181 · f98f181
1 parent fe83cc7
commit f98f181
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cleanrl/ppo.py b/cleanrl/ppo.py
@@ -81,7 +81,7 @@ class Args:
 def make_env(env_id, idx, capture_video, run_name):
     def thunk():
         if capture_video and idx == 0:
-            env = gym.make(env_id)
+            env = gym.make(env_id, render_mode="rgb_array")
             env = gym.wrappers.RecordVideo(env, f"videos/{run_name}")
         else:
             env = gym.make(env_id)