fancy_gym/alr_envs/open_ai/fetch/positional_wrapper.py

13 lines
444 B
Python
Raw Normal View History

2021-07-01 14:55:14 +02:00
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]
2021-07-01 14:55:14 +02:00
@property
def current_pos(self) -> Union[float, int, np.ndarray]:
return self.unwrapped._get_obs()["observation"][:4]