Mc fix (#40)

* Mc fix * Apply Black code formatting Co-authored-by: Slin Lee <slin@pathmind.com>
PathmindAI · Nov 4, 2021 · 31e6d49 · 31e6d49
1 parent f7d123c
commit 31e6d49
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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):
         """