finish examples 2 and 3

2024-03-26 16:13:40 +00:00
parent cb6d1e0871
commit 1ed0110900
10 changed files with 196 additions and 42 deletions
--- a/lerobot/common/datasets/pusht.py
+++ b/lerobot/common/datasets/pusht.py
@@ -100,6 +100,7 @@ class PushtDataset(AbstractDataset):
        collate_fn: Callable | None = None,
        writer: Writer | None = None,
        transform: "torchrl.envs.Transform" = None,
+        dummy: bool = False,
    ):
        super().__init__(
            dataset_id,
@@ -113,6 +114,7 @@ class PushtDataset(AbstractDataset):
            collate_fn=collate_fn,
            writer=writer,
            transform=transform,
+            dummy=dummy,
        )

    def _download_and_preproc_obsolete(self):