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
Download Patch File Download Diff File Expand all files Collapse all files
fancy_gym/black_box/black_box_wrapper.py
+2 -1
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()