Adopt new interface structure

This commit is contained in:
Marcel
2021-07-05 09:16:36 +02:00
parent 37c4ab8a41
commit 0046ade102
10 changed files with 25 additions and 53 deletions
-1
View File
@@ -1,2 +1 @@
from alr_envs.open_ai.reacher_v2.positional_wrapper import PositionalWrapper
from alr_envs.open_ai.reacher_v2.mp_wrapper import MPWrapper
+7 -2
View File
@@ -1,13 +1,18 @@
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 self.sim.data.qvel[:2]
@property
def current_pos(self) -> Union[float, int, np.ndarray]:
return self.sim.data.qpos[:2]
@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 self.sim.data.qvel[:2]
@property
def current_pos(self) -> Union[float, int, np.ndarray]:
return self.sim.data.qpos[:2]