From 671cde51ca30d26d3b1480de1e39eae0c84090fc Mon Sep 17 00:00:00 2001
From: number_cruncher <lennart.oestreich@stud.tu-darmstadt.de>
Date: Sun, 16 Mar 2025 19:07:56 +0100
Subject: [PATCH] rgb

---
 a2c_sb3_cartpole.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/a2c_sb3_cartpole.py b/a2c_sb3_cartpole.py
index 7245853..7c94adc 100644
--- a/a2c_sb3_cartpole.py
+++ b/a2c_sb3_cartpole.py
@@ -20,7 +20,7 @@ run = wandb.init(
     save_code=True,
 )
 
-env = gym.make("CartPole-v1")
+env = gym.make("CartPole-v1", render_mode="rgb_array")
 
 model = A2C("MlpPolicy", env, verbose=1, tensorboard_log=f"runs/{run.id}")
 #model = A2C("MlpPolicy", env, )
-- 
GitLab