From dada377135dfa96cbcfd1beea02ecc4559d97829 Mon Sep 17 00:00:00 2001
From: Bjarne-55 <73470930+Bjarne-55@users.noreply.github.com>
Date: Tue, 15 Oct 2024 21:25:37 +0200
Subject: [PATCH] Reset_all now returns real info dictionary

---
 mushroom_rl/environments/isaac_env.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/mushroom_rl/environments/isaac_env.py b/mushroom_rl/environments/isaac_env.py
index b3bd14bc..ec94f158 100644
--- a/mushroom_rl/environments/isaac_env.py
+++ b/mushroom_rl/environments/isaac_env.py
@@ -90,8 +90,9 @@ def reset_all(self, env_mask, state=None):
             self._task.reset_idx(idxs)
         # self._world.step(render=self._render) # TODO Check if we can do otherwise
         task_obs = self._task.get_observations()
+        task_extras = self._task.get_extras()
         observation = convert_task_observation(task_obs)
-        return observation.clone(), [{}]*self._n_envs
+        return observation.clone(), [task_extras]*self._n_envs
 
     def step_all(self, env_mask, action):
         self._task.pre_physics_step(action)