update ppo.py

vwxyzjn · Oct 15, 2023 · d00d87c · d00d87c
1 parent d1d28c7
commit d00d87c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cleanrl/ppo.py b/cleanrl/ppo.py
@@ -79,7 +79,7 @@ def parse_args():
 def make_env(env_id, seed, idx, capture_video, run_name):
     def thunk():
         if capture_video and idx == 0:
-            env = gym.make(env_id, render_mode="rgb_array")
+            env = gym.make(env_id)
             env = gym.wrappers.RecordVideo(env, f"videos/{run_name}")
         else:
             env = gym.make(env_id)