Add LeRobotDatasetMetadata

2024-11-03 18:07:37 +01:00
parent ac79e8cb36
commit e4ba084e25
25 changed files with 419 additions and 327 deletions
--- a/lerobot/common/datasets/utils.py
+++ b/lerobot/common/datasets/utils.py
@@ -139,7 +139,7 @@ def load_tasks(local_dir: Path) -> dict:
    return {item["task_index"]: item["task"] for item in sorted(tasks, key=lambda x: x["task_index"])}


-def load_episode_dicts(local_dir: Path) -> dict:
+def load_episodes(local_dir: Path) -> dict:
    return load_jsonlines(local_dir / EPISODES_PATH)