update infos of box pushing envs

fancy_gym/black_box/black_box_wrapper.py

@@ -47,6 +47,8 @@ class BlackBoxWrapper(gym.ObservationWrapper):
 
         # trajectory generation
         self.traj_gen = trajectory_generator
+
+        self.traj_gen.basis_gn.show_basis(plot=True)
         self.tracking_controller = tracking_controller
         # self.time_steps = np.linspace(0, self.duration, self.traj_steps)
         # self.traj_gen.set_mp_times(self.time_steps)
@@ -159,7 +161,7 @@ class BlackBoxWrapper(gym.ObservationWrapper):
                                                 t + 1 + self.current_traj_steps):
                 break
 
-        infos.update({k: v[:t] for k, v in infos.items()})
+        infos.update({k: v[:t+1] for k, v in infos.items()})
         self.current_traj_steps += t + 1
 
         if self.verbose >= 2: