diff --git a/pathmind/simulation.py b/pathmind/simulation.py index d174bc1..0b0ccd1 100644 --- a/pathmind/simulation.py +++ b/pathmind/simulation.py @@ -132,6 +132,8 @@ def run( for episode in range(num_episodes): step = 0 + done = False + self.reset() while not done: row = [episode, step] if sleep: @@ -169,6 +171,7 @@ def run( write_table(table=table, out_csv=out_csv) write_table(table=summary, out_csv=summary_csv) + print(f"--------Finished episode {episode}--------") def train(self, base_folder: str = "./", observation_yaml: str = None): """