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