diff --git a/metastable_baselines/misc/rollout_buffer.py b/metastable_baselines/misc/rollout_buffer.py index 2b68ffe..85eed07 100644 --- a/metastable_baselines/misc/rollout_buffer.py +++ b/metastable_baselines/misc/rollout_buffer.py @@ -280,4 +280,4 @@ class GaussianRolloutCollectorAuxclass(): def get_past_trajectories(self): # TODO: Respect Episode Boundaries - return self.actions + return self.rollout_buffer.actions