dodox/fancy_gym
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fixed Bug: env_id was broken, when prefix is handled by other library

Browse Source 
(e.g. 'dmc:...' by shimmy)
This commit is contained in:
Dominik Moritz Roth 2023-05-18 19:08:11 +02:00
parent d4a8306b9d
commit 0dfd8e18e4
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
fancy_gym/utils/make_env_helpers.py
View File

@ -77,13 +77,13 @@ def make(env_id: str, seed: int, **kwargs):
if ':' in env_id: if ':' in env_id:
split_id = env_id.split(':') split_id = env_id.split(':')
framework, env_id = split_id[-2:] framework, framework_env_id = split_id[-2:]
else: else:
framework = None framework = None
if framework == 'metaworld': if framework == 'metaworld':
# MetaWorld environment # MetaWorld environment
env = make_metaworld(env_id, seed, **kwargs) env = make_metaworld(framework_env_id, seed, **kwargs)
# elif framework == 'dmc': # elif framework == 'dmc':
# Deprecated: With shimmy gym now has native support for deepmind envs # Deprecated: With shimmy gym now has native support for deepmind envs
# # DeepMind Control environment # # DeepMind Control environment