Adopt new interface structure

2021-07-05 09:16:36 +02:00
parent 37c4ab8a41
commit 0046ade102
10 changed files with 25 additions and 53 deletions
@@ -1,2 +1 @@
-from alr_envs.open_ai.continuous_mountain_car.positional_wrapper import PositionalWrapper
 from alr_envs.open_ai.continuous_mountain_car.mp_wrapper import MPWrapper
@@ -1,12 +1,17 @@
 from typing import Union

+import numpy as np
 from mp_env_api.env_wrappers.mp_env_wrapper import MPEnvWrapper


 class MPWrapper(MPEnvWrapper):
    @property
-    def start_pos(self):
-        raise ValueError("Start position is not available")
+    def current_vel(self) -> Union[float, int, np.ndarray]:
+        return np.array([self.state[1]])
+
+    @property
+    def current_pos(self) -> Union[float, int, np.ndarray]:
+        return np.array([self.state[0]])

    @property
    def goal_pos(self):
@@ -1,13 +0,0 @@
-from typing import Union
-import numpy as np
-from mp_env_api.env_wrappers.positional_env_wrapper import PositionalEnvWrapper
-
-
-class PositionalWrapper(PositionalEnvWrapper):
-    @property
-    def current_vel(self) -> Union[float, int, np.ndarray]:
-        return np.array([self.state[1]])
-
-    @property
-    def current_pos(self) -> Union[float, int, np.ndarray]:
-        return np.array([self.state[0]])