This commit is contained in:
ottofabian 2021-06-25 15:43:24 +02:00
parent 3c774382b7
commit 29b8c3a6c7
2 changed files with 2 additions and 2 deletions

View File

@ -5,7 +5,7 @@ import numpy as np
from mp_env_api.envs.mp_env_wrapper import MPEnvWrapper from mp_env_api.envs.mp_env_wrapper import MPEnvWrapper
class ALRBallInACupMPWrapper(MPEnvWrapper): class BallInACupMPWrapper(MPEnvWrapper):
@property @property
def active_obs(self): def active_obs(self):

View File

@ -5,7 +5,7 @@ import numpy as np
from mp_env_api.envs.positional_env_wrapper import PositionalEnvWrapper from mp_env_api.envs.positional_env_wrapper import PositionalEnvWrapper
class BallInACupPositionalEnvWrapper(PositionalEnvWrapper): class BallInACupPositionalWrapper(PositionalEnvWrapper):
@property @property
def current_pos(self) -> Union[float, int, np.ndarray, Tuple]: def current_pos(self) -> Union[float, int, np.ndarray, Tuple]:
return self.sim.data.qpos[0:7].copy() return self.sim.data.qpos[0:7].copy()