dodox/fancy_gym
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Bug fixes for wrapped OpenAi environments

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
alr_envs/__init__.py
View File

@ -571,7 +571,7 @@ register(
"mp_kwargs": { "mp_kwargs": {
"num_dof": 1, "num_dof": 1,
"num_basis": 4, "num_basis": 4,
"duration": 100, "duration": 2,
"post_traj_time": 0, "post_traj_time": 0,
"width": 0.02, "width": 0.02,
"policy_type": "motor", "policy_type": "motor",
@ -613,7 +613,7 @@ register(
"mp_kwargs": { "mp_kwargs": {
"num_dof": 4, "num_dof": 4,
"num_basis": 5, "num_basis": 5,
"duration": 1, "duration": 2,
"post_traj_time": 0, "post_traj_time": 0,
"width": 0.02, "width": 0.02,
"policy_type": "position" "policy_type": "position"
@ -630,7 +630,7 @@ register(
"mp_kwargs": { "mp_kwargs": {
"num_dof": 4, "num_dof": 4,
"num_basis": 5, "num_basis": 5,
"duration": 1, "duration": 2,
"post_traj_time": 0, "post_traj_time": 0,
"width": 0.02, "width": 0.02,
"policy_type": "position" "policy_type": "position"

2
alr_envs/open_ai/continuous_mountain_car/mp_wrapper.py
View File

@ -14,4 +14,4 @@ class MPWrapper(MPEnvWrapper):
@property @property
def dt(self) -> Union[float, int]: def dt(self) -> Union[float, int]:
return 1. return 0.02

4
alr_envs/open_ai/fetch/positional_wrapper.py
View File

@ -6,8 +6,8 @@ from mp_env_api.env_wrappers.positional_env_wrapper import PositionalEnvWrapper
class PositionalWrapper(PositionalEnvWrapper): class PositionalWrapper(PositionalEnvWrapper):
@property @property
def current_vel(self) -> Union[float, int, np.ndarray]: 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 @property
def current_pos(self) -> Union[float, int, np.ndarray]: def current_pos(self) -> Union[float, int, np.ndarray]:
return self._get_obs()["observation"][:4] return self.unwrapped._get_obs()["observation"][:4]