Merged PR 62: Make reward calculation consider red POL

Check out the linked bug ticket to understand the issue. The fix was very simple- just changing which variable is passed to the reward calculation funciton. Related work items: #1442
2023-06-06 12:27:55 +00:00
parent 4a09673e98 bb42b4f55d
commit 3c9061ff8f
1 changed files with 1 additions and 1 deletions
--- a/src/primaite/environment/primaite_env.py
+++ b/src/primaite/environment/primaite_env.py
@@ -368,7 +368,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,