19 lines
454 B
Python
19 lines
454 B
Python
from typing import Union
|
|
|
|
import numpy as np
|
|
from mp_env_api.interface_wrappers.mp_env_wrapper import MPEnvWrapper
|
|
|
|
|
|
class MPWrapper(MPEnvWrapper):
|
|
|
|
@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]
|
|
|
|
@property
|
|
def dt(self) -> Union[float, int]:
|
|
return self.env.dt |