Fix: Did not define MP_Wrapper for new RandomInit version of BoxPushing

This commit is contained in:
Dominik Moritz Roth 2023-10-11 13:29:08 +02:00
parent 614d2a42db
commit f8dcf5b2df

View File

@ -215,6 +215,7 @@ for reward_type in ["Dense", "TemporalSparse", "TemporalSpatialSparse"]:
register(
id='fancy/BoxPushingRandomInit{}-v0'.format(reward_type),
entry_point='fancy_gym.envs.mujoco:BoxPushing{}'.format(reward_type),
mp_wrapper=mujoco.box_pushing.MPWrapper,
max_episode_steps=MAX_EPISODE_STEPS_BOX_PUSHING,
kwargs={"random_init": True}
)