oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove temporary print statements

Browse Source
This commit is contained in:
Marek Wolan
2024-01-10 09:07:51 +00:00
parent e73783f6fa
commit b7cc940e9d
2 changed files with 0 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/primaite/game/agent/data_manipulation_bot.py
View File

@@ -15,8 +15,6 @@ class DataManipulationAgent(AbstractScriptedAgent):
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
super().__init__(*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) self._set_next_execution_timestep(self.agent_settings.start_settings.start_step)
def _set_next_execution_timestep(self, timestep: int) -> None: 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.agent_settings.start_settings.variance, self.agent_settings.start_settings.variance
) )
self.next_execution_timestep = timestep + random_timestep_increment 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]: def get_action(self, obs: ObsType, reward: float = None) -> Tuple[str, Dict]:
"""Randomly sample an action from the action space. """Randomly sample an action from the action space.
@@ -43,12 +40,10 @@ class DataManipulationAgent(AbstractScriptedAgent):
current_timestep = self.action_manager.game.step_counter current_timestep = self.action_manager.game.step_counter
if current_timestep < self.next_execution_timestep: if current_timestep < self.next_execution_timestep:
print("red agent doing nothing")
return "DONOTHING", {"dummy": 0} return "DONOTHING", {"dummy": 0}
self._set_next_execution_timestep(current_timestep + self.agent_settings.start_settings.frequency) 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} return "NODE_APPLICATION_EXECUTE", {"node_id": 0, "application_id": 0}
def reset_agent_for_episode(self) -> None: def reset_agent_for_episode(self) -> None:

1
src/primaite/game/agent/rewards.py
View File

@@ -181,7 +181,6 @@ class WebServer404Penalty(AbstractReward):
""" """
web_service_state = access_from_nested_dict(state, self.location_in_state) web_service_state = access_from_nested_dict(state, self.location_in_state)
if web_service_state is NOT_PRESENT_IN_STATE: if web_service_state is NOT_PRESENT_IN_STATE:
print("error getting web service state")
return 0.0 return 0.0
most_recent_return_code = web_service_state["last_response_status_code"] 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. # TODO: reward needs to use the current web state. Observation should return web state at the time of last scan.