Allow dmc envs to be accessed via dmc:... (for backwards compatibility)

This commit is contained in:
Dominik Moritz Roth 2023-05-18 19:50:19 +02:00
parent 21d5ebb442
commit 07de655025

View File

@ -84,10 +84,11 @@ def make(env_id: str, seed: int, **kwargs):
if framework == 'metaworld':
# MetaWorld environment
env = make_metaworld(framework_env_id, seed, **kwargs)
# elif framework == 'dmc':
# Deprecated: With shimmy gym now has native support for deepmind envs
# # DeepMind Control environment
# env = make_dmc(env_id, seed, **kwargs)
elif framework == 'dmc':
# DeepMind Control environment
# ensures legacy compatability:
# shimmy expects dm_controll/..., while we used dmc:... in the past
env = make_gym('dm_control/'+framework_env_id, seed, **kwargs)
else:
env = make_gym(env_id, seed, **kwargs)