Apply suggestions from code review

This commit is contained in:
Christopher McCarthy
2023-11-23 15:45:22 +00:00
parent efeaa4c1cc
commit 76939fb8e8

View File

@@ -281,7 +281,7 @@ class PrimaiteSession:
self.episode_counter += 1
self.step_counter = 0
_LOGGER.debug(f"Restting primaite session, episode = {self.episode_counter}")
self.simulation = self._simulation_initial_state
self.simulation = copy.deepcopy(self._simulation_initial_state)
def close(self) -> None:
"""Close the session, this will stop the env and close the simulation."""