From 778361cd6b811d7777e926e14f04d7ffa35b680b Mon Sep 17 00:00:00 2001
From: Bjarne-55 <73470930+Bjarne-55@users.noreply.github.com>
Date: Tue, 15 Oct 2024 23:25:47 +0200
Subject: [PATCH] Parse info and episode_info before returning

---
 mushroom_rl/core/core.py | 2 ++
 1 file changed, 2 insertions(+)

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):