Bug fixes for wrapped OpenAi environments

This commit is contained in:
Marcel
2021-07-05 08:14:39 +02:00
parent 585bdc7b47
commit 28d10ef089
3 changed files with 6 additions and 6 deletions
+2 -2
View File
@@ -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]