From f4fdef2f0321a9caeac72bee7260eb78d4fbb8a7 Mon Sep 17 00:00:00 2001 From: Antonio Serrano Munoz Date: Wed, 14 Aug 2024 15:45:03 -0400 Subject: [PATCH] Apply some changes from multi-agent branch to avoid merge conflict --- skrl/envs/wrappers/torch/__init__.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/skrl/envs/wrappers/torch/__init__.py b/skrl/envs/wrappers/torch/__init__.py index abcbff60..b4176bbd 100644 --- a/skrl/envs/wrappers/torch/__init__.py +++ b/skrl/envs/wrappers/torch/__init__.py @@ -104,8 +104,8 @@ def _in(values, container): logger.info(f"Environment wrapper: 'auto' (class: {', '.join(base_classes)})") if _in("omni.isaac.lab.envs..*", base_classes): - 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 "isaaclab" + elif _in("omni.isaac.gym..*", base_classes): return "omniverse-isaacgym" elif _in(["isaacgymenvs..*", "tasks..*.VecTask"], base_classes): return "isaacgym-preview4" # preview 4 is the same as 3 @@ -113,7 +113,7 @@ def _in(values, container): return "isaacgym-preview2" elif _in("robosuite.environments.", base_classes): return "robosuite" - elif _in("dm_env._environment.Environment.", base_classes): + elif _in("dm_env..*", base_classes): return "dm" elif _in("pettingzoo.utils.env", base_classes) or _in("pettingzoo.utils.wrappers", base_classes): return "pettingzoo"