From b7cc940e9dd1485b4aada9d0f208bb17b3c4bee4 Mon Sep 17 00:00:00 2001
From: Marek Wolan <marek.wolan@methods.co.uk>
Date: Wed, 10 Jan 2024 09:07:51 +0000
Subject: [PATCH] Remove temporary print statements

---
 src/primaite/game/agent/data_manipulation_bot.py | 5 -----
 src/primaite/game/agent/rewards.py               | 1 -
 2 files changed, 6 deletions(-)

diff --git a/src/primaite/game/agent/data_manipulation_bot.py b/src/primaite/game/agent/data_manipulation_bot.py
index 3b558087..7ad45518 100644
--- a/src/primaite/game/agent/data_manipulation_bot.py
+++ b/src/primaite/game/agent/data_manipulation_bot.py
@@ -15,8 +15,6 @@ class DataManipulationAgent(AbstractScriptedAgent):
 
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
-        print("red start step: ", self.agent_settings.start_settings.start_step)
-
         self._set_next_execution_timestep(self.agent_settings.start_settings.start_step)
 
     def _set_next_execution_timestep(self, timestep: int) -> None:
@@ -28,7 +26,6 @@ class DataManipulationAgent(AbstractScriptedAgent):
             -self.agent_settings.start_settings.variance, self.agent_settings.start_settings.variance
         )
         self.next_execution_timestep = timestep + random_timestep_increment
-        print("next execution red step: ", self.next_execution_timestep)
 
     def get_action(self, obs: ObsType, reward: float = None) -> Tuple[str, Dict]:
         """Randomly sample an action from the action space.
@@ -43,12 +40,10 @@ class DataManipulationAgent(AbstractScriptedAgent):
         current_timestep = self.action_manager.game.step_counter
 
         if current_timestep < self.next_execution_timestep:
-            print("red agent doing nothing")
             return "DONOTHING", {"dummy": 0}
 
         self._set_next_execution_timestep(current_timestep + self.agent_settings.start_settings.frequency)
 
-        print("red agent doing an execute")
         return "NODE_APPLICATION_EXECUTE", {"node_id": 0, "application_id": 0}
 
     def reset_agent_for_episode(self) -> None:
diff --git a/src/primaite/game/agent/rewards.py b/src/primaite/game/agent/rewards.py
index 9b3dfb80..cb8f8cb1 100644
--- a/src/primaite/game/agent/rewards.py
+++ b/src/primaite/game/agent/rewards.py
@@ -181,7 +181,6 @@ class WebServer404Penalty(AbstractReward):
         """
         web_service_state = access_from_nested_dict(state, self.location_in_state)
         if web_service_state is NOT_PRESENT_IN_STATE:
-            print("error getting web service state")
             return 0.0
         most_recent_return_code = web_service_state["last_response_status_code"]
         # TODO: reward needs to use the current web state. Observation should return web state at the time of last scan.