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

Remove distracting debug print statements

Browse Source
This commit is contained in:
Marek Wolan
2023-12-01 11:07:57 +00:00
parent deea09f88d
commit 3642e87eda
3 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/primaite/session/policy/sb3.py
View File

@@ -51,14 +51,13 @@ class SB3Policy(PolicyABC, identifier="SB3"):
def eval(self, n_episodes: int, deterministic: bool) -> None: def eval(self, n_episodes: int, deterministic: bool) -> None:
"""Evaluate the agent.""" """Evaluate the agent."""
reward_data = evaluate_policy( _ = evaluate_policy(
self._agent, self._agent,
self.session.env, self.session.env,
n_eval_episodes=n_episodes, n_eval_episodes=n_episodes,
deterministic=deterministic, deterministic=deterministic,
return_episode_rewards=True, return_episode_rewards=True,
) )
print(reward_data)
def save(self, save_path: Path) -> None: def save(self, save_path: Path) -> None:
""" """

2
src/primaite/simulator/system/services/database/database_service.py
View File

@@ -56,7 +56,7 @@ class DatabaseService(Service):
def reset_component_for_episode(self, episode: int): def reset_component_for_episode(self, episode: int):
"""Reset the original state of the SimComponent.""" """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() self.connections.clear()
super().reset_component_for_episode(episode) super().reset_component_for_episode(episode)

1
src/primaite/simulator/system/services/web_server/web_server.py
View File

@@ -47,7 +47,6 @@ class WebServer(Service):
state["last_response_status_code"] = ( state["last_response_status_code"] = (
self.last_response_status_code.value if isinstance(self.last_response_status_code, HttpStatusCode) else None self.last_response_status_code.value if isinstance(self.last_response_status_code, HttpStatusCode) else None
) )
print(state)
return state return state
def __init__(self, **kwargs): def __init__(self, **kwargs):