update test suite for replanning envs

2022-11-14 17:39:46 +01:00
parent 7e3ec7a2ef
commit be14b21fff
3 changed files with 167 additions and 30 deletions
@@ -86,6 +86,7 @@ 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