wrappers updated

alr_envs/alr/mujoco/ant_jump/mp_wrapper.py

@@ -2,13 +2,13 @@ from typing import Tuple, Union
 
 import numpy as np
 
-from mp_env_api import MPEnvWrapper
+from alr_envs.mp.raw_interface_wrapper import RawInterfaceWrapper
 
 
-class MPWrapper(MPEnvWrapper):
+class MPWrapper(RawInterfaceWrapper):
 
     @property
-    def active_obs(self):
+    def context_mask(self) -> np.ndarray:
         return np.hstack([
             [False] * 111, # ant has 111 dimensional observation space !!
             [True] # goal height