Fixed wrong env name (still contained mp info)

This commit is contained in:
Dominik Moritz Roth 2023-09-18 19:18:26 +02:00
parent 5488ca6dda
commit 725e6c8fe1

View File

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