minor updates

2022-11-14 17:48:15 +01:00
parent be14b21fff
commit fc3051bf57
3 changed files with 7 additions and 10 deletions
@@ -86,7 +86,6 @@ class BlackBoxWrapper(gym.ObservationWrapper):
        return observation.astype(self.observation_space.dtype)

    def get_trajectory(self, action: np.ndarray) -> Tuple:
-        # duration = self.duration - self.current_traj_steps * self.dt
        duration = self.duration
        if self.learn_sub_trajectories:
            duration = None