13 lines
444 B
Python
13 lines
444 B
Python
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.unwrapped._get_obs()["observation"][-5:-1]
|
|
|
|
@property
|
|
def current_pos(self) -> Union[float, int, np.ndarray]:
|
|
return self.unwrapped._get_obs()["observation"][:4] |