Add agent reset for episodes

This commit is contained in:
Marek Wolan
2024-01-09 14:53:15 +00:00
parent f2a496893c
commit daa34385e5
3 changed files with 14 additions and 0 deletions

View File

@@ -162,6 +162,7 @@ class PrimaiteGame:
self.simulation.reset_component_for_episode(episode=self.episode_counter)
for agent in self.agents:
agent.reward_function.total_reward = 0.0
agent.reset_agent_for_episode()
def close(self) -> None:
"""Close the game, this will close the simulation."""