Parse info and episode_info before returning

MushroomRL · Oct 15, 2024 · 778361c · 778361c
1 parent f587fab
commit 778361c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mushroom_rl/core/core.py b/mushroom_rl/core/core.py
@@ -128,6 +128,8 @@ def _run(self, dataset, n_steps, n_episodes, render, quiet, record, initial_stat
 
         self._end(record)
 
+        dataset.info.parse()
+        dataset.episode_info.parse()
         return dataset
 
     def _step(self, render, record):