From 0f72a5e4780e8a85266e9ec6b1a501c996d1b380 Mon Sep 17 00:00:00 2001
From: Phil <s8phsaue@stud.uni-saarland.de>
Date: Fri, 7 Mar 2025 14:31:21 +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 23f6cb2..e1bb8b1 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 truncated
+                    state, action, reward, cost, next_state, not terminated
                 )
             
             # End episode if the cost exceeded the limit
-- 
GitLab