wrappers updated

alr_envs/open_ai/classic_control/continuous_mountain_car/mp_wrapper.py

@@ -1,10 +1,11 @@
 from typing import 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 current_vel(self) -> Union[float, int, np.ndarray]:
         return np.array([self.state[1]])

alr_envs/open_ai/mujoco/reacher_v2/mp_wrapper.py

@@ -1,10 +1,11 @@
 from typing import 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 current_vel(self) -> Union[float, int, np.ndarray]:
@@ -13,7 +14,3 @@ class MPWrapper(MPEnvWrapper):
     @property
     def current_pos(self) -> Union[float, int, np.ndarray]:
         return self.sim.data.qpos[:2]
-
-    @property
-    def dt(self) -> Union[float, int]:
-        return self.env.dt

alr_envs/open_ai/robotics/fetch/mp_wrapper.py

@@ -2,10 +2,10 @@ from typing import 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):