eod commit

2024-02-23 10:06:48 +00:00
parent 72f4cc0a50
commit f933341df5
6 changed files with 24 additions and 30 deletions
--- a/src/primaite/simulator/sim_container.py
+++ b/src/primaite/simulator/sim_container.py
@@ -21,9 +21,9 @@ class Simulation(SimComponent):

        super().__init__(**kwargs)

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

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