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

Fixed typo in env registration

Browse Source
This commit is contained in:
Dominik Moritz Roth 2023-09-17 17:17:20 +02:00
parent 315e135ff0
commit 7f95923cf4
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'dmc_cartpole-{_task}_dmp-v0', id=f'dm_control/cartpole-{_task}_dmp-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'],