Bug fixes for wrapped OpenAi environments
This commit is contained in:
@@ -6,8 +6,8 @@ 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._get_obs()["observation"][-5:-1]
|
||||
return self.unwrapped._get_obs()["observation"][-5:-1]
|
||||
|
||||
@property
|
||||
def current_pos(self) -> Union[float, int, np.ndarray]:
|
||||
return self._get_obs()["observation"][:4]
|
||||
return self.unwrapped._get_obs()["observation"][:4]
|
||||
Reference in New Issue
Block a user