From beaf5df2fcd7d957ed1f5c678ab6dcaa0a2215ce Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Thu, 23 Nov 2023 21:14:11 +0100 Subject: [PATCH] Bring to parity with dev-branch --- test/test_all_gym_builtin_envs.py | 4 +++- test/test_black_box.py | 2 +- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/test/test_all_gym_builtin_envs.py b/test/test_all_gym_builtin_envs.py index f2eeac6..9264b98 100644 --- a/test/test_all_gym_builtin_envs.py +++ b/test/test_all_gym_builtin_envs.py @@ -12,7 +12,9 @@ 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 'shimmy' not in spec.id.lower() + and 'ale_py' not in spec.id.lower() + and 'tabular' 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'] diff --git a/test/test_black_box.py b/test/test_black_box.py index c1a760a..fc4f0bf 100644 --- a/test/test_black_box.py +++ b/test/test_black_box.py @@ -103,7 +103,7 @@ def test_verbosity(mp_type: str, env_wrap: Tuple[str, Type[RawInterfaceWrapper]] env_step = make(env_id) env_step.reset() - _obs, _reward, _terminated, _truncated, info = env.step(env.action_space.sample()) + _obs, _reward, _terminated, _truncated, info = env_step.step(env_step.action_space.sample()) info_keys_step = info.keys() assert all(e in info_keys for e in info_keys_step)