wrappers updated

alr_envs/alr/mujoco/ball_in_a_cup/ball_in_a_cup_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 BallInACupMPWrapper(MPEnvWrapper):
+class BallInACupMPWrapper(RawInterfaceWrapper):
 
     @property
-    def active_obs(self):
+    def context_mask(self) -> np.ndarray:
         # TODO: @Max Filter observations correctly
         return np.hstack([
             [False] * 7,  # cos