From 77a6fd6aff8fae9740cd1f0c5852938135c7ff8b Mon Sep 17 00:00:00 2001 From: Marek Wolan Date: Tue, 30 May 2023 08:50:57 +0000 Subject: [PATCH] Make reward calculation consider red POL --- src/primaite/environment/primaite_env.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/primaite/environment/primaite_env.py b/src/primaite/environment/primaite_env.py index 99c7c09f..3fe7f0f6 100644 --- a/src/primaite/environment/primaite_env.py +++ b/src/primaite/environment/primaite_env.py @@ -398,7 +398,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,