Compare commits
3 Commits
master
...
fix_test_b
Author | SHA1 | Date | |
---|---|---|---|
c8b79d276d | |||
b4128ff6d9 | |||
50664d953b |
@ -103,7 +103,7 @@ def test_verbosity(mp_type: str, env_wrap: Tuple[str, Type[RawInterfaceWrapper]]
|
|||||||
|
|
||||||
env_step = make(env_id)
|
env_step = make(env_id)
|
||||||
env_step.reset()
|
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()
|
info_keys_step = info.keys()
|
||||||
|
|
||||||
assert all(e in info_keys for e in info_keys_step)
|
assert all(e in info_keys for e in info_keys_step)
|
||||||
|
@ -12,13 +12,14 @@ GYM_IDS = [spec.id for spec in gym.envs.registry.values() if
|
|||||||
not isinstance(spec.entry_point, Callable) and
|
not isinstance(spec.entry_point, Callable) and
|
||||||
"fancy_gym" not in spec.entry_point and 'make_bb_env_helper' not in spec.entry_point
|
"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 '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)
|
and not re.match(r'GymV2.Environment', spec.id)
|
||||||
]
|
]
|
||||||
GYM_MP_IDS = fancy_gym.ALL_DMC_MOVEMENT_PRIMITIVE_ENVIRONMENTS['all']
|
GYM_MP_IDS = fancy_gym.ALL_DMC_MOVEMENT_PRIMITIVE_ENVIRONMENTS['all']
|
||||||
SEED = 1
|
SEED = 1
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize('env_id', GYM_IDS)
|
@pytest.mark.parametrize('env_id', GYM_IDS)
|
||||||
def test_step_gym_functionality(env_id: str):
|
def test_step_gym_functionality(env_id: str):
|
||||||
"""Tests that step environments run without errors using random actions."""
|
"""Tests that step environments run without errors using random actions."""
|
Loading…
Reference in New Issue
Block a user