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:
dodox
2023-05-21 17:29:03 +02:00
parent ea7634ca42
commit d4003e1a68
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
metastable_baselines/misc/rollout_buffer.py
+2 -2
View File
@@ -278,6 +278,6 @@ class GaussianRolloutCollectorAuxclass():
return True return True
def get_past_trajectories(self): def get_past_trajectories(self) -> th.Tensor:
# TODO: Respect Episode Boundaries # TODO: Respect Episode Boundaries
return self.rollout_buffer.actions return th.Tensor(self.rollout_buffer.actions)