Merge branch 'toni/fix_gymnasium_unwrapped_warning' into toni/test_en…

…v_wrappers
Toni-SM · Aug 4, 2024 · a22d843 · a22d843
2 parents 15cac8f + 80169ce
commit a22d843
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/skrl/envs/wrappers/torch/isaaclab_envs.py b/skrl/envs/wrappers/torch/isaaclab_envs.py
@@ -1,4 +1,4 @@
-from typing import Any, Tuple
+from typing import Any, Tuple, Union
 
 import gymnasium
 
@@ -21,13 +21,17 @@ def __init__(self, env: Any) -> None:
         self._info = {}
 
     @property
-    def state_space(self) -> gymnasium.Space:
+    def state_space(self) -> Union[gymnasium.Space, None]:
         """State space
         """
         try:
             return self._unwrapped.single_observation_space["critic"]
-        except:
+        except KeyError:
+            pass
+        try:
             return self._unwrapped.state_space
+        except AttributeError:
+            return None
 
     @property
     def observation_space(self) -> gymnasium.Space: