Merge remote-tracking branch 'origin/dev' into feature/893-node-acl-into-one-action-space

2023-06-06 13:46:01 +01:00
parent 2e1bdf2361 3c9061ff8f
commit d572c033e1
1 changed files with 1 additions and 1 deletions
--- a/src/primaite/environment/primaite_env.py
+++ b/src/primaite/environment/primaite_env.py
@@ -361,7 +361,7 @@ class Primaite(Env):
        # 5. Calculate reward signal (for RL)
        reward = calculate_reward_function(
            self.nodes_post_pol,
-            self.nodes_post_blue,
+            self.nodes_post_red,
            self.nodes_reference,
            self.green_iers,
            self.red_iers,