Skip to content

Commit

Permalink
Apply some changes from multi-agent branch to avoid merge conflict
Browse files Browse the repository at this point in the history
  • Loading branch information
Toni-SM committed Aug 14, 2024
1 parent f104915 commit f4fdef2
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions skrl/envs/wrappers/torch/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -104,16 +104,16 @@ 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
elif _in("rlgpu.tasks..*.VecTask", base_classes):
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"
Expand Down

0 comments on commit f4fdef2

Please sign in to comment.