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

Use new namespace convention for mp-versions of envs

Browse Source
This commit is contained in:
Dominik Moritz Roth 2023-08-14 10:51:43 +02:00
parent 14223c56ba
commit fb1282fe1e
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
fancy_gym/envs/registry.py
View File

@ -167,9 +167,11 @@ def register_mp(id, mp_wrapper, mp_type, mp_config_override={}):
else: else:
raise ValueError('env id can not contain multiple "/".') raise ValueError('env id can not contain multiple "/".')
parts = id.split('-') parts = name.split('-')
assert len(parts) >= 2 and parts[-1].startswith('v'), 'Malformed env id, must end in -v{int}.' assert len(parts) >= 2 and parts[-1].startswith('v'), 'Malformed env id, must end in -v{int}.'
fancy_id = '-'.join(parts[:-1]+[mp_type, parts[-1]]) fancy_name = '-'.join(parts[:-1]+[mp_type, parts[-1]])
fancy_id = f'{ns}_{mp_type}/{fancy_name}'
gym_register( gym_register(
id=fancy_id, id=fancy_id,