Refactor datasets into LeRobotDataset (#91)

Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
2024-04-25 12:23:12 +02:00
parent e760e4cd63
commit 659c69a1c0
90 changed files with 167 additions and 352 deletions
--- a/lerobot/configs/env/aloha.yaml
+++ b/lerobot/configs/env/aloha.yaml
@@ -10,7 +10,8 @@ online_steps: 25000

 fps: 50

-dataset_id: aloha_sim_insertion_human
+dataset:
+  repo_id: lerobot/aloha_sim_insertion_human

 env:
  name: aloha
--- a/lerobot/configs/env/pusht.yaml
+++ b/lerobot/configs/env/pusht.yaml
@@ -10,7 +10,8 @@ online_steps: 25000

 fps: 10

-dataset_id: pusht
+dataset:
+  repo_id: lerobot/pusht

 env:
  name: pusht
--- a/lerobot/configs/env/xarm.yaml
+++ b/lerobot/configs/env/xarm.yaml
@@ -9,7 +9,8 @@ online_steps: 25000

 fps: 15

-dataset_id: xarm_lift_medium
+dataset:
+  repo_id: lerobot/xarm_lift_medium

 env:
  name: xarm