dodox/metastable-baselines
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

get_past_trajectories will now return a th.Tensor

Browse Source
This commit is contained in:
Dominik Moritz Roth 2023-05-21 17:29:03 +02:00
parent ea7634ca42
commit d4003e1a68
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
metastable_baselines/misc/rollout_buffer.py
View File

@ -278,6 +278,6 @@ class GaussianRolloutCollectorAuxclass():
return True
def get_past_trajectories(self):
def get_past_trajectories(self) -> th.Tensor:
# TODO: Respect Episode Boundaries
return self.rollout_buffer.actions
return th.Tensor(self.rollout_buffer.actions)