Set rgb array render mode in sb3_highway_dqn_cnn.py

Fix #566
Farama-Foundation · Feb 4, 2024 · 3966526 · 3966526
1 parent 81c11d6
commit 3966526
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/sb3_highway_dqn_cnn.py b/scripts/sb3_highway_dqn_cnn.py
@@ -6,7 +6,7 @@
 
 
 def train_env():
-    env = gym.make("highway-fast-v0")
+    env = gym.make("highway-fast-v0", render_mode="rgb_array")
     env.configure(
         {
             "observation": {