Dont run determinism tests on arbitrary external envs added by other

libs (like atari envs added by shimmy), only those included in gym and those added by us.
This commit is contained in:
Dominik Moritz Roth 2023-10-11 13:08:06 +02:00
parent 877a7ea6ba
commit 600575cbac

View File

@ -12,6 +12,7 @@ GYM_IDS = [spec.id for spec in gym.envs.registry.values() if
not isinstance(spec.entry_point, Callable) and
"fancy_gym" not in spec.entry_point and 'make_bb_env_helper' not in spec.entry_point
and 'jax' not in spec.id.lower()
and 'jax' not in spec.id.lower()
and not re.match(r'GymV2.Environment', spec.id)
]
GYM_MP_IDS = fancy_gym.ALL_DMC_MOVEMENT_PRIMITIVE_ENVIRONMENTS['all']