diff --git a/src/lerobot/datasets/lerobot_dataset.py b/src/lerobot/datasets/lerobot_dataset.py index 99dbf606e..979453623 100644 --- a/src/lerobot/datasets/lerobot_dataset.py +++ b/src/lerobot/datasets/lerobot_dataset.py @@ -104,9 +104,6 @@ class LeRobotDatasetMetadata: self.revision = get_safe_version(self.repo_id, self.revision) (self.root / "meta").mkdir(exist_ok=True, parents=True) - # TODO(rcadene): instead of downloading all episodes metadata files, - # download only the ones associated to the requested episodes. This would - # require adding `episodes: list[int]` as argument. self.pull_from_repo(allow_patterns="meta/") self.load_metadata() diff --git a/tests/fixtures/dataset_factories.py b/tests/fixtures/dataset_factories.py index 783058014..c33fdcb72 100644 --- a/tests/fixtures/dataset_factories.py +++ b/tests/fixtures/dataset_factories.py @@ -54,7 +54,6 @@ class LeRobotDatasetFactory(Protocol): def get_task_index(tasks: datasets.Dataset, task: str) -> int: - # TODO(rcadene): a bit complicated no? ^^ task_idx = tasks.loc[task].task_index.item() return task_idx