Finalise the refactor. It works well now.

2024-02-25 17:44:41 +00:00
parent a34cf08209
commit 994dbc3501
10 changed files with 65 additions and 54 deletions
--- a/src/primaite/simulator/sim_container.py
+++ b/src/primaite/simulator/sim_container.py
@@ -23,7 +23,7 @@ class Simulation(SimComponent):

    def setup_for_episode(self, episode: int):
        """Reset the original state of the SimComponent."""
-        self.network.setup_for_episode(episode)
+        self.network.setup_for_episode(episode=episode)

    def _init_request_manager(self) -> RequestManager:
        rm = super()._init_request_manager()