Reset_all now returns real info dictionary

MushroomRL · Oct 15, 2024 · dada377 · dada377
1 parent 550971f
commit dada377
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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)