Make environment reset reinstantiate the game

This commit is contained in:
Marek Wolan
2024-02-20 16:21:03 +00:00
parent 88f8e9cb42
commit 64b9ba3ecf
3 changed files with 21 additions and 15 deletions

View File

@@ -67,9 +67,6 @@ class PrimaiteGame:
self.step_counter: int = 0
"""Current timestep within the episode."""
self.episode_counter: int = 0
"""Current episode number."""
self.options: PrimaiteGameOptions
"""Special options that apply for the entire game."""
@@ -163,7 +160,7 @@ class PrimaiteGame:
return True
return False
def reset(self) -> None:
def reset(self) -> None: # TODO: deprecated - remove me
"""Reset the game, this will reset the simulation."""
self.episode_counter += 1
self.step_counter = 0