Fix auto wrapping for Isaac Gym preview 4 in torch

Toni-SM · Aug 5, 2024 · d2edbcf · d2edbcf
1 parent a6e93d0
commit d2edbcf
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/skrl/envs/wrappers/torch/__init__.py b/skrl/envs/wrappers/torch/__init__.py
@@ -91,7 +91,9 @@ def _in(values, container):
             return "isaaclab"
         elif _in("omni.isaac.gym.vec_env.vec_env_base.VecEnvBase", base_classes) or _in("omni.isaac.gym.vec_env.vec_env_mt.VecEnvMT", base_classes):
             return "omniverse-isaacgym"
-        elif _in("rlgpu.tasks.base.vec_task.VecTask", base_classes):
+        elif _in("tasks..*.VecTask", base_classes):
+            return "isaacgym-preview4"  # preview 4 is the same as 3
+        elif _in("rlgpu.tasks..*.VecTask", base_classes):
             return "isaacgym-preview2"
         elif _in("robosuite.environments.", base_classes):
             return "robosuite"