adjust env registries in __init__

alr_envs/alr/mujoco/hopper_jump/__init__.py

@@ -1,2 +1 @@
-from .mp_wrapper import MPWrapper, HighCtxtMPWrapper
-from .new_mp_wrapper import NewMPWrapper, NewHighCtxtMPWrapper
+from .new_mp_wrapper import MPWrapper

alr_envs/alr/mujoco/hopper_jump/new_mp_wrapper.py

@@ -3,7 +3,7 @@ from typing import Union, Tuple
 import numpy as np
 
 
-class NewMPWrapper(BlackBoxWrapper):
+class MPWrapper(BlackBoxWrapper):
     @property
     def current_pos(self) -> Union[float, int, np.ndarray, Tuple]:
         return self.env.sim.data.qpos[3:6].copy()
@@ -30,7 +30,7 @@ class NewMPWrapper(BlackBoxWrapper):
             ])
 
 
-class NewHighCtxtMPWrapper(NewMPWrapper):
+class NewHighCtxtMPWrapper(MPWrapper):
     def get_context_mask(self):
         return np.hstack([
             [False] * (2 + int(not self.env.exclude_current_positions_from_observation)),  # position