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

Fixed Bug leading to infinite recusrion when making mp envs

Browse Source
This commit is contained in:
Dominik Moritz Roth 2023-07-23 11:06:08 +02:00
parent ce34df1814
commit 21a9c0f8a3
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/envs/registry.py
View File

@ -7,7 +7,7 @@ from fancy_gym.utils.make_env_helpers import make_bb
from fancy_gym.utils.utils import nested_update from fancy_gym.utils.utils import nested_update
from gymnasium import register as gym_register from gymnasium import register as gym_register
from gymnasium import gym_make from gymnasium import make as gym_make
from fancy_gym.black_box.raw_interface_wrapper import RawInterfaceWrapper from fancy_gym.black_box.raw_interface_wrapper import RawInterfaceWrapper
@ -135,7 +135,7 @@ def register_mp(id, mp_wrapper, mp_type, mp_config_override={}):
parts = id.split('-') parts = id.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_id = '-'.join(parts[:-1]+[mp_type, parts[-1]])
register( gym_register(
id=fancy_id, id=fancy_id,
entry_point=bb_env_constructor, entry_point=bb_env_constructor,
kwargs={ kwargs={