updated test to pytest

2022-09-26 09:46:53 +02:00
parent 756a9fb039
commit 39815ad09a
10 changed files with 133 additions and 497 deletions
@@ -77,7 +77,6 @@ class BlackBoxWrapper(gym.ObservationWrapper):
        # TODO: is this correct for replanning? Do we need to adjust anything here?
        bc_time = np.array(0 if not self.do_replanning else self.current_traj_steps * self.dt)
        self.traj_gen.set_boundary_conditions(bc_time, self.current_pos, self.current_vel)
-        # TODO: remove the - self.dt after Bruces fix.
        self.traj_gen.set_duration(None if self.learn_sub_trajectories else self.duration, self.dt)
        # traj_dict = self.traj_gen.get_trajs(get_pos=True, get_vel=True)
        trajectory = get_numpy(self.traj_gen.get_traj_pos())