From 39665261ba04f83ceff4a4eb04df2b9994357b05 Mon Sep 17 00:00:00 2001
From: Edouard Leurent <eleurent@gmail.com>
Date: Sun, 4 Feb 2024 19:40:32 +0000
Subject: [PATCH] Set rgb array render mode in sb3_highway_dqn_cnn.py

Fix https://github.com/Farama-Foundation/HighwayEnv/issues/566
---
 scripts/sb3_highway_dqn_cnn.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/sb3_highway_dqn_cnn.py b/scripts/sb3_highway_dqn_cnn.py
index 057fa71b1..8c9116013 100644
--- 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": {