added new mp wrappers to all environments

This commit is contained in:
Onur
2022-06-29 10:39:28 +02:00
parent 02b8a65bab
commit 6e06e11cfa
9 changed files with 180 additions and 5 deletions
@@ -10,7 +10,7 @@ class MPWrapper(MPEnvWrapper):
def active_obs(self):
return np.hstack([
[False] * 17,
[True] # goal pos
[True] # goal pos
])
@property
@@ -0,0 +1,27 @@
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 MPWrapper(RawInterfaceWrapper):
def context_mask(self):
return np.hstack([
[False] * 17,
[True] # goal pos
])
@property
def current_pos(self) -> Union[float, int, np.ndarray]:
return self.env.sim.data.qpos[3:6].copy()
@property
def current_vel(self) -> Union[float, int, np.ndarray, Tuple]:
return self.env.sim.data.qvel[3:6].copy()
@property
def dt(self) -> Union[float, int]:
return self.env.dt