From f98f181282cb4562bd47d6c7ea1103497a11d8fe Mon Sep 17 00:00:00 2001
From: Costa Huang <costa.huang@outlook.com>
Date: Mon, 27 Nov 2023 22:02:34 -0500
Subject: [PATCH] quick fix

---
 cleanrl/ppo.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cleanrl/ppo.py b/cleanrl/ppo.py
index bfec99563..edde76bae 100644
--- 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)