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

use observation mask for replanning

Browse Source
This commit is contained in:
Hongyi Zhou 2022-10-31 13:52:40 +01:00
parent 61c1b76e29
commit cd2b679e9b
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
fancy_gym/black_box/black_box_wrapper.py
View File

@ -61,7 +61,8 @@ class BlackBoxWrapper(gym.ObservationWrapper):
# self.traj_gen.basis_gn.show_basis(plot=True)
# spaces
self.return_context_observation = not (learn_sub_trajectories or self.do_replanning)
# self.return_context_observation = not (learn_sub_trajectories or self.do_replanning)
self.return_context_observation = True
self.traj_gen_action_space = self._get_traj_gen_action_space()
# self.action_space = self._get_action_space()

2
fancy_gym/envs/mujoco/box_pushing/mp_wrapper.py
View File

@ -20,7 +20,7 @@ class MPWrapper(RawInterfaceWrapper):
[True] * 4, # orientation of box
[True] * 3, # position of target
[True] * 4, # orientation of target
# [True] * 1, # time
[True] * 1, # time
])
@property