IsaacLabWrapper reset method bug

[shubhs0007]

Description

info variable is undefined if self._reset_once is False

skrl/skrl/envs/wrappers/torch/isaaclab_envs.py

Line 43 in 636936f

return self._obs_dict["policy"], info

Resulting in the following error -

UnboundLocalError: local variable 'info' referenced before assignment

What skrl version are you using?

1.2.0

What ML framework/library version are you using?

torch 2.2.2+cu118

Additional system information

Python 3.10.14, Ubuntu

[bekleyis95]

I created an MR for this issue in here.

[Toni-SM]

Hi @shubhs0007 and @bekleyis95

Solved in the toni/update_wrappers branch with also include some test for it.

skrl/skrl/envs/wrappers/torch/isaaclab_envs.py

Line 63 in 9e5b5e4

self._obs_dict, reward, terminated, truncated, self._info = self._env.step(actions)

skrl/skrl/envs/wrappers/torch/isaaclab_envs.py

Line 73 in 9e5b5e4

self._obs_dict, self._info = self._env.reset()