From cffdcdc0d21184f1622e711fb820b368927146df Mon Sep 17 00:00:00 2001 From: Marek Wolan Date: Tue, 27 Jun 2023 15:27:56 +0100 Subject: [PATCH] Fix ier reward calculation --- src/primaite/environment/reward.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/primaite/environment/reward.py b/src/primaite/environment/reward.py index aa9e4503..1a1a0770 100644 --- a/src/primaite/environment/reward.py +++ b/src/primaite/environment/reward.py @@ -78,8 +78,8 @@ def calculate_reward_function( start_step = ier_value.get_start_step() stop_step = ier_value.get_end_step() if step_count >= start_step and step_count <= stop_step: - reference_blocked = reference_ier.get_is_running() - live_blocked = ier_value.get_is_running() + reference_blocked = not reference_ier.get_is_running() + live_blocked = not ier_value.get_is_running() ier_reward = ( config_values.green_ier_blocked * ier_value.get_mission_criticality() )