Skip to content

Commit dada377

Browse files
committed
Reset_all now returns real info dictionary
1 parent 550971f commit dada377

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

mushroom_rl/environments/isaac_env.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -90,8 +90,9 @@ def reset_all(self, env_mask, state=None):
9090
self._task.reset_idx(idxs)
9191
# self._world.step(render=self._render) # TODO Check if we can do otherwise
9292
task_obs = self._task.get_observations()
93+
task_extras = self._task.get_extras()
9394
observation = convert_task_observation(task_obs)
94-
return observation.clone(), [{}]*self._n_envs
95+
return observation.clone(), [task_extras]*self._n_envs
9596

9697
def step_all(self, env_mask, action):
9798
self._task.pre_physics_step(action)

0 commit comments

Comments
 (0)