Reintroduce last step

huawei-noah · Mar 1, 2022 · f345c71 · f345c71
1 parent a9465f7
commit f345c71
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/smarts/env/wrappers/episode_logger.py b/smarts/env/wrappers/episode_logger.py
@@ -56,3 +56,4 @@ def _episode_logs(self, col_width) -> EpisodeLog:
         with EpisodeLogs(col_width) as episode_logs:
             while not self._done:
                 yield episode_logs.reset()
+            episode_logs.reset()