Skip to content

Commit 6e38458

Browse files
authored
[BugFix] Fix wrong split_trajectories import (#3023)
1 parent 350fa1d commit 6e38458

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchrl/data/datasets/minari_data.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -350,7 +350,7 @@ def _download_and_preproc(self):
350350
# Add a "done" entry
351351
if self.split_trajs:
352352
with td_data.unlock_():
353-
from torchrl.objectives.utils import split_trajectories
353+
from torchrl.collectors.utils import split_trajectories
354354

355355
td_data = split_trajectories(td_data).memmap_(self.data_path)
356356
with open(self.metadata_path, "w") as metadata_file:

0 commit comments

Comments
 (0)