updated setup.py

2022-12-05 14:07:45 +01:00
parent 46824f5d41
commit 5cbefb2b11
3 changed files with 11 additions and 7 deletions
@@ -166,7 +166,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: