diff --git a/mushroom_rl/core/core.py b/mushroom_rl/core/core.py index 3dffe448..3478029c 100644 --- 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):