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

shorter duration

Browse Source
This commit is contained in:
Hongyi Zhou 2022-11-22 11:13:08 +01:00
parent be1a9bd67e
commit a219f90d72
1 changed files with 2 additions and 1 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

@ -99,7 +99,8 @@ class BlackBoxWrapper(gym.ObservationWrapper):
def get_trajectory(self, action: np.ndarray) -> Tuple:
# duration = self.duration
duration = self.duration - self.current_traj_steps * self.dt
# duration = self.duration - self.current_traj_steps * self.dt
duration = action[0]
if self.learn_sub_trajectories:
duration = None
# reset with every new call as we need to set all arguments, such as tau, delay, again.