Support passing trajectories into predictiion functions (for pca)

2023-05-21 20:16:25 +02:00
parent d4003e1a68
commit 9cd25ee484
3 changed files with 84 additions and 4 deletions
@@ -280,4 +280,4 @@ class GaussianRolloutCollectorAuxclass():

    def get_past_trajectories(self) -> th.Tensor:
        # TODO: Respect Episode Boundaries
-        return th.Tensor(self.rollout_buffer.actions)
+        return np.swapaxes(th.Tensor(self.rollout_buffer.actions[:self.rollout_buffer.pos]), 0, 1)