From a7c0658262664a9a01515ea8d064fcc28a768160 Mon Sep 17 00:00:00 2001
From: Phil <s8phsaue@stud.uni-saarland.de>
Date: Fri, 7 Mar 2025 14:32:03 +0100
Subject: [PATCH] Bugfix done mask

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

diff --git a/main.py b/main.py
index e1bb8b1..bff7053 100644
--- a/main.py
+++ b/main.py
@@ -163,7 +163,7 @@ def run_single_exploration(args, env, agent:CSCCQLSAC, buffer:ReplayBuffer, stat
             # During training, add experiences to buffer
             if train:
                 buffer.add(
-                    state, action, reward, cost, next_state, not terminated
+                    state, action, reward, cost, next_state, terminated
                 )
             
             # End episode if the cost exceeded the limit
-- 
GitLab