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

Fix: Don't throw errors for envs, that do not expose a max_episode_steps.

Browse Source
This commit is contained in:
Dominik Moritz Roth 2023-06-18 11:53:10 +02:00
parent f44f01b478
commit 49ac9c3785
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
test/test_black_box.py
View File

@ -115,6 +115,11 @@ def test_verbosity(mp_type: str, env_wrap: Tuple[str, Type[RawInterfaceWrapper]]
@pytest.mark.parametrize('mp_type', ['promp', 'dmp', 'prodmp']) @pytest.mark.parametrize('mp_type', ['promp', 'dmp', 'prodmp'])
@pytest.mark.parametrize('env_wrap', zip(ENV_IDS, WRAPPERS)) @pytest.mark.parametrize('env_wrap', zip(ENV_IDS, WRAPPERS))
def test_length(mp_type: str, env_wrap: Tuple[str, Type[RawInterfaceWrapper]]): def test_length(mp_type: str, env_wrap: Tuple[str, Type[RawInterfaceWrapper]]):
if not env.spec.max_episode_steps:
# Not all envs expose a max_episode_steps.
# To use those with MPs, they could be put in a time_limit-wrapper.
return True
basis_generator_type = 'prodmp' if mp_type == 'prodmp' else 'rbf' basis_generator_type = 'prodmp' if mp_type == 'prodmp' else 'rbf'
env_id, wrapper_class = env_wrap env_id, wrapper_class = env_wrap