small fix in type + comments

2024-03-15 12:44:52 +00:00
parent 41521f7e96
commit 5805a7ffb1
5 changed files with 10 additions and 7 deletions
--- a/lerobot/common/datasets/aloha.py
+++ b/lerobot/common/datasets/aloha.py
@@ -87,7 +87,7 @@ class AlohaExperienceReplay(AbstractExperienceReplay):
        batch_size: int = None,
        *,
        shuffle: bool = True,
-        root: Path = None,
+        root: Path | None = None,
        pin_memory: bool = False,
        prefetch: int = None,
        sampler: SliceSampler = None,