Call base class's close method and rename internal variable

skrl/envs/wrappers/torch/isaaclab_envs.py

@@ -17,7 +17,7 @@ def __init__(self, env: Any) -> None:
         super().__init__(env)
 
         self._reset_once = True
-        self._obs_dict = None
+        self._observations = None
         self._info = {}
 
     @property
@@ -60,8 +60,8 @@ def step(self, actions: torch.Tensor) -> Tuple[torch.Tensor, torch.Tensor, torch
         :return: Observation, reward, terminated, truncated, info
         :rtype: tuple of torch.Tensor and any other info
         """
-        self._obs_dict, reward, terminated, truncated, self._info = self._env.step(actions)
-        return self._obs_dict["policy"], reward.view(-1, 1), terminated.view(-1, 1), truncated.view(-1, 1), self._info
+        self._observations, reward, terminated, truncated, self._info = self._env.step(actions)
+        return self._observations["policy"], reward.view(-1, 1), terminated.view(-1, 1), truncated.view(-1, 1), self._info
 
     def reset(self) -> Tuple[torch.Tensor, Any]:
         """Reset the environment
@@ -70,16 +70,11 @@ def reset(self) -> Tuple[torch.Tensor, Any]:
         :rtype: torch.Tensor and any other info
         """
         if self._reset_once:
-            self._obs_dict, self._info = self._env.reset()
+            self._observations, self._info = self._env.reset()
             self._reset_once = False
-        return self._obs_dict["policy"], self._info
+        return self._observations["policy"], self._info
 
     def render(self, *args, **kwargs) -> None:
         """Render the environment
         """
         pass
-
-    def close(self) -> None:
-        """Close the environment
-        """
-        self._env.close()