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

Fixed wrong env name (still contained mp info)

Browse Source
This commit is contained in:
Dominik Moritz Roth 2023-09-18 19:18:26 +02:00
parent 5488ca6dda
commit 725e6c8fe1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
fancy_gym/dmc/__init__.py
View File

@ -32,7 +32,7 @@ register(
_dmc_cartpole_tasks = ["balance", "balance_sparse", "swingup", "swingup_sparse"] _dmc_cartpole_tasks = ["balance", "balance_sparse", "swingup", "swingup_sparse"]
for _task in _dmc_cartpole_tasks: for _task in _dmc_cartpole_tasks:
register( register(
id=f'dm_control/cartpole-{_task}_dmp-v0', id=f'dm_control/cartpole-{_task}-v0',
register_step_based=False, register_step_based=False,
mp_wrapper=suite.cartpole.MPWrapper, mp_wrapper=suite.cartpole.MPWrapper,
add_mp_types=['DMP', 'ProMP'], add_mp_types=['DMP', 'ProMP'],