This commit is contained in:
Alexander Soare
2024-03-27 18:33:48 +00:00
parent 120f0aef5c
commit b7c9c33072
10 changed files with 20 additions and 33 deletions

View File

@@ -100,7 +100,6 @@ class PushtDataset(AbstractDataset):
collate_fn: Callable | None = None,
writer: Writer | None = None,
transform: "torchrl.envs.Transform" = None,
dummy: bool = False,
):
super().__init__(
dataset_id,
@@ -114,7 +113,6 @@ class PushtDataset(AbstractDataset):
collate_fn=collate_fn,
writer=writer,
transform=transform,
dummy=dummy,
)
def _download_and_preproc_obsolete(self):