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

fix action space bugs for bp

Browse Source
This commit is contained in:
Onur
2022-05-03 21:34:39 +02:00
parent 2fbde9fbb1
commit 640f3b2d90
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
alr_envs/alr/mujoco/beerpong/new_mp_wrapper.py
+2 -2
View File
@@ -38,7 +38,7 @@ class NewMPWrapper(EpisodicWrapper):
min_action_bounds, max_action_bounds = self.mp.get_param_bounds()
min_action_bounds = np.concatenate((min_action_bounds.numpy(), [self.env.action_space.low[-1]]))
max_action_bounds = np.concatenate((max_action_bounds.numpy(), [self.env.action_space.high[-1]]))
self.mp_action_space = gym.spaces.Box(low=min_action_bounds, high=max_action_bounds, dtype=np.float32)
return self.mp_action_space
self.action_space = gym.spaces.Box(low=min_action_bounds, high=max_action_bounds, dtype=np.float32)
return self.action_space
else:
return super(NewMPWrapper, self).set_action_space()