From f8dcf5b2df86952ca1549ebd258855c22cea28d1 Mon Sep 17 00:00:00 2001
From: Dominik Roth <dominik.roth.dev@gmail.com>
Date: Wed, 11 Oct 2023 13:29:08 +0200
Subject: [PATCH] Fix: Did not define MP_Wrapper for new RandomInit version of
 BoxPushing

---
 fancy_gym/envs/__init__.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/fancy_gym/envs/__init__.py b/fancy_gym/envs/__init__.py
index d416dbb..a40c81f 100644
--- a/fancy_gym/envs/__init__.py
+++ b/fancy_gym/envs/__init__.py
@@ -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}
     )