From 3642e87eda51f2317a15dad7908d78f2b5246a10 Mon Sep 17 00:00:00 2001
From: Marek Wolan <marek.wolan@methods.co.uk>
Date: Fri, 1 Dec 2023 11:07:57 +0000
Subject: [PATCH] Remove distracting debug print statements

---
 src/primaite/session/policy/sb3.py                             | 3 +--
 .../simulator/system/services/database/database_service.py     | 2 +-
 .../simulator/system/services/web_server/web_server.py         | 1 -
 3 files changed, 2 insertions(+), 4 deletions(-)

diff --git a/src/primaite/session/policy/sb3.py b/src/primaite/session/policy/sb3.py
index 051e2770..254baf4d 100644
--- a/src/primaite/session/policy/sb3.py
+++ b/src/primaite/session/policy/sb3.py
@@ -51,14 +51,13 @@ class SB3Policy(PolicyABC, identifier="SB3"):
 
     def eval(self, n_episodes: int, deterministic: bool) -> None:
         """Evaluate the agent."""
-        reward_data = evaluate_policy(
+        _ = evaluate_policy(
             self._agent,
             self.session.env,
             n_eval_episodes=n_episodes,
             deterministic=deterministic,
             return_episode_rewards=True,
         )
-        print(reward_data)
 
     def save(self, save_path: Path) -> None:
         """
diff --git a/src/primaite/simulator/system/services/database/database_service.py b/src/primaite/simulator/system/services/database/database_service.py
index f9621ba5..bba4e777 100644
--- a/src/primaite/simulator/system/services/database/database_service.py
+++ b/src/primaite/simulator/system/services/database/database_service.py
@@ -56,7 +56,7 @@ class DatabaseService(Service):
 
     def reset_component_for_episode(self, episode: int):
         """Reset the original state of the SimComponent."""
-        print("Resetting DatabaseService original state on node {self.software_manager.node.hostname}")
+        _LOGGER.debug("Resetting DatabaseService original state on node {self.software_manager.node.hostname}")
         self.connections.clear()
         super().reset_component_for_episode(episode)
 
diff --git a/src/primaite/simulator/system/services/web_server/web_server.py b/src/primaite/simulator/system/services/web_server/web_server.py
index bff29a47..e5f3dccc 100644
--- a/src/primaite/simulator/system/services/web_server/web_server.py
+++ b/src/primaite/simulator/system/services/web_server/web_server.py
@@ -47,7 +47,6 @@ class WebServer(Service):
         state["last_response_status_code"] = (
             self.last_response_status_code.value if isinstance(self.last_response_status_code, HttpStatusCode) else None
         )
-        print(state)
         return state
 
     def __init__(self, **kwargs):