Include observation.environment_state with keypoints in PushT dataset (#303)

Co-authored-by: Remi <re.cadene@gmail.com>
2024-07-09 08:27:40 +01:00
parent 7bd5ab16d1
commit a4d77b99f0
27 changed files with 739 additions and 609 deletions
--- a/lerobot/init.py
+++ b/lerobot/init.py
@@ -70,6 +70,8 @@ available_datasets_per_env = {
        "lerobot/aloha_sim_transfer_cube_human_image",
        "lerobot/aloha_sim_transfer_cube_scripted_image",
    ],
+    # TODO(alexander-soare): Add "lerobot/pusht_keypoints". Right now we can't because this is too tightly
+    # coupled with tests.
    "pusht": ["lerobot/pusht", "lerobot/pusht_image"],
    "xarm": [
        "lerobot/xarm_lift_medium",