chore(dataset): add check dataset shape

Merge branch 'main' into fix/save-episode-zero-width-feature
fix(record): connect teleoperator before robot to avoid watchdog jump (#4166 )
2026-07-28 12:15:59 +00:00 · 2026-07-27 18:36:15 +02:00 · 2026-07-27 15:52:12 +02:00 · 2026-07-27 14:09:58 +02:00 · 2026-07-27 14:08:11 +02:00 · 2026-07-27 13:44:32 +02:00
20 changed files with 262 additions and 140 deletions
@@ -72,19 +72,19 @@ jobs:
      HF_USER_TOKEN: ${{ secrets.LEROBOT_HF_USER }}

    steps:
-      - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1  # v7.0.1
+      - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
        with:
          persist-credentials: false
          lfs: true

      - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v4 # zizmor: ignore[unpinned-uses]
+        uses: docker/setup-buildx-action@v3 # zizmor: ignore[unpinned-uses]
        with:
          cache-binary: false

      - name: Login to Docker Hub
        if: ${{ env.DOCKERHUB_USERNAME != '' }}
-        uses: docker/login-action@v4.4.0 # zizmor: ignore[unpinned-uses]
+        uses: docker/login-action@v3 # zizmor: ignore[unpinned-uses]
        with:
          username: ${{ secrets.DOCKERHUB_LEROBOT_USERNAME }}
          password: ${{ secrets.DOCKERHUB_LEROBOT_PASSWORD }}
@@ -95,7 +95,7 @@ jobs:
      # from source-copy, so code-only changes skip the slow uv-sync layer
      # when the runner has a warm Docker daemon cache.
      - name: Build Libero benchmark image
-        uses: docker/build-push-action@v7 # zizmor: ignore[unpinned-uses]
+        uses: docker/build-push-action@v6 # zizmor: ignore[unpinned-uses]
        with:
          context: .
          file: docker/Dockerfile.benchmark.libero
@@ -151,7 +151,7 @@ jobs:

      - name: Upload Libero rollout video
        if: always()
-        uses: actions/upload-artifact@v7 # zizmor: ignore[unpinned-uses]
+        uses: actions/upload-artifact@v4 # zizmor: ignore[unpinned-uses]
        with:
          name: libero-rollout-video
          path: /tmp/libero-artifacts/videos/
@@ -159,7 +159,7 @@ jobs:

      - name: Upload Libero eval metrics
        if: always()
-        uses: actions/upload-artifact@v7 # zizmor: ignore[unpinned-uses]
+        uses: actions/upload-artifact@v4 # zizmor: ignore[unpinned-uses]
        with:
          name: libero-metrics
          path: /tmp/libero-artifacts/metrics.json
@@ -214,7 +214,7 @@ jobs:

      - name: Upload Libero train-smoke eval video
        if: always()
-        uses: actions/upload-artifact@v7 # zizmor: ignore[unpinned-uses]
+        uses: actions/upload-artifact@v4 # zizmor: ignore[unpinned-uses]
        with:
          name: libero-train-smoke-video
          path: /tmp/libero-train-smoke-artifacts/eval/
@@ -230,19 +230,19 @@ jobs:
      HF_USER_TOKEN: ${{ secrets.LEROBOT_HF_USER }}

    steps:
-      - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1  # v7.0.1
+      - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
        with:
          persist-credentials: false
          lfs: true

      - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v4 # zizmor: ignore[unpinned-uses]
+        uses: docker/setup-buildx-action@v3 # zizmor: ignore[unpinned-uses]
        with:
          cache-binary: false

      - name: Login to Docker Hub
        if: ${{ env.DOCKERHUB_USERNAME != '' }}
-        uses: docker/login-action@v4.4.0 # zizmor: ignore[unpinned-uses]
+        uses: docker/login-action@v3 # zizmor: ignore[unpinned-uses]
        with:
          username: ${{ secrets.DOCKERHUB_LEROBOT_USERNAME }}
          password: ${{ secrets.DOCKERHUB_LEROBOT_PASSWORD }}
@@ -250,7 +250,7 @@ jobs:
          DOCKERHUB_USERNAME: ${{ secrets.DOCKERHUB_LEROBOT_USERNAME }}

      - name: Build MetaWorld benchmark image
-        uses: docker/build-push-action@v7 # zizmor: ignore[unpinned-uses]
+        uses: docker/build-push-action@v6 # zizmor: ignore[unpinned-uses]
        with:
          context: .
          file: docker/Dockerfile.benchmark.metaworld
@@ -303,7 +303,7 @@ jobs:

      - name: Upload MetaWorld rollout video
        if: always()
-        uses: actions/upload-artifact@v7 # zizmor: ignore[unpinned-uses]
+        uses: actions/upload-artifact@v4 # zizmor: ignore[unpinned-uses]
        with:
          name: metaworld-rollout-video
          path: /tmp/metaworld-artifacts/videos/
@@ -311,7 +311,7 @@ jobs:

      - name: Upload MetaWorld eval metrics
        if: always()
-        uses: actions/upload-artifact@v7 # zizmor: ignore[unpinned-uses]
+        uses: actions/upload-artifact@v4 # zizmor: ignore[unpinned-uses]
        with:
          name: metaworld-metrics
          path: /tmp/metaworld-artifacts/metrics.json
@@ -332,19 +332,19 @@ jobs:
      ROBOTWIN_TASKS: beat_block_hammer,click_bell,handover_block,stack_blocks_two,click_alarmclock,open_microwave,adjust_bottle,lift_pot,stamp_seal,turn_switch

    steps:
-      - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1  # v7.0.1
+      - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
        with:
          persist-credentials: false
          lfs: true

      - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v4 # zizmor: ignore[unpinned-uses]
+        uses: docker/setup-buildx-action@v3 # zizmor: ignore[unpinned-uses]
        with:
          cache-binary: false

      - name: Login to Docker Hub
        if: ${{ env.DOCKERHUB_USERNAME != '' }}
-        uses: docker/login-action@v4.4.0 # zizmor: ignore[unpinned-uses]
+        uses: docker/login-action@v3 # zizmor: ignore[unpinned-uses]
        with:
          username: ${{ secrets.DOCKERHUB_LEROBOT_USERNAME }}
          password: ${{ secrets.DOCKERHUB_LEROBOT_PASSWORD }}
@@ -355,7 +355,7 @@ jobs:
      # simulation assets (~4 GB). Layer cache lives in the runner's local
      # Docker daemon — reused across re-runs on the same machine.
      - name: Build RoboTwin 2.0 benchmark image
-        uses: docker/build-push-action@v7 # zizmor: ignore[unpinned-uses]
+        uses: docker/build-push-action@v6 # zizmor: ignore[unpinned-uses]
        with:
          context: .
          file: docker/Dockerfile.benchmark.robotwin
@@ -413,7 +413,7 @@ jobs:

      - name: Upload RoboTwin rollout video
        if: always()
-        uses: actions/upload-artifact@v7
+        uses: actions/upload-artifact@v4
        with:
          name: robotwin-rollout-video
          path: /tmp/robotwin-artifacts/videos/
@@ -421,7 +421,7 @@ jobs:

      - name: Upload RoboTwin eval metrics
        if: always()
-        uses: actions/upload-artifact@v7
+        uses: actions/upload-artifact@v4
        with:
          name: robotwin-metrics
          path: /tmp/robotwin-artifacts/metrics.json
@@ -439,19 +439,19 @@ jobs:
      HF_USER_TOKEN: ${{ secrets.LEROBOT_HF_USER }}

    steps:
-      - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1  # v7.0.1
+      - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
        with:
          persist-credentials: false
          lfs: true

      - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v4 # zizmor: ignore[unpinned-uses]
+        uses: docker/setup-buildx-action@v3 # zizmor: ignore[unpinned-uses]
        with:
          cache-binary: false

      - name: Login to Docker Hub
        if: ${{ env.DOCKERHUB_USERNAME != '' }}
-        uses: docker/login-action@v4.4.0 # zizmor: ignore[unpinned-uses]
+        uses: docker/login-action@v3 # zizmor: ignore[unpinned-uses]
        with:
          username: ${{ secrets.DOCKERHUB_LEROBOT_USERNAME }}
          password: ${{ secrets.DOCKERHUB_LEROBOT_PASSWORD }}
@@ -459,7 +459,7 @@ jobs:
          DOCKERHUB_USERNAME: ${{ secrets.DOCKERHUB_LEROBOT_USERNAME }}

      - name: Build RoboCasa365 benchmark image
-        uses: docker/build-push-action@v7 # zizmor: ignore[unpinned-uses]
+        uses: docker/build-push-action@v6 # zizmor: ignore[unpinned-uses]
        with:
          context: .
          file: docker/Dockerfile.benchmark.robocasa
@@ -514,7 +514,7 @@ jobs:

      - name: Upload RoboCasa365 rollout video
        if: always()
-        uses: actions/upload-artifact@v7 # zizmor: ignore[unpinned-uses]
+        uses: actions/upload-artifact@v4 # zizmor: ignore[unpinned-uses]
        with:
          name: robocasa-rollout-video
          path: /tmp/robocasa-artifacts/videos/
@@ -522,7 +522,7 @@ jobs:

      - name: Upload RoboCasa365 eval metrics
        if: always()
-        uses: actions/upload-artifact@v7 # zizmor: ignore[unpinned-uses]
+        uses: actions/upload-artifact@v4 # zizmor: ignore[unpinned-uses]
        with:
          name: robocasa-metrics
          path: /tmp/robocasa-artifacts/metrics.json
@@ -540,19 +540,19 @@ jobs:
      HF_USER_TOKEN: ${{ secrets.LEROBOT_HF_USER }}

    steps:
-      - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1  # v7.0.1
+      - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
        with:
          persist-credentials: false
          lfs: true

      - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v4 # zizmor: ignore[unpinned-uses]
+        uses: docker/setup-buildx-action@v3 # zizmor: ignore[unpinned-uses]
        with:
          cache-binary: false

      - name: Login to Docker Hub
        if: ${{ env.DOCKERHUB_USERNAME != '' }}
-        uses: docker/login-action@v4.4.0 # zizmor: ignore[unpinned-uses]
+        uses: docker/login-action@v3 # zizmor: ignore[unpinned-uses]
        with:
          username: ${{ secrets.DOCKERHUB_LEROBOT_USERNAME }}
          password: ${{ secrets.DOCKERHUB_LEROBOT_PASSWORD }}
@@ -560,7 +560,7 @@ jobs:
          DOCKERHUB_USERNAME: ${{ secrets.DOCKERHUB_LEROBOT_USERNAME }}

      - name: Build RoboCerebra benchmark image
-        uses: docker/build-push-action@v7 # zizmor: ignore[unpinned-uses]
+        uses: docker/build-push-action@v6 # zizmor: ignore[unpinned-uses]
        with:
          context: .
          file: docker/Dockerfile.benchmark.robocerebra
@@ -621,7 +621,7 @@ jobs:

      - name: Upload RoboCerebra rollout video
        if: always()
-        uses: actions/upload-artifact@v7 # zizmor: ignore[unpinned-uses]
+        uses: actions/upload-artifact@v4 # zizmor: ignore[unpinned-uses]
        with:
          name: robocerebra-rollout-video
          path: /tmp/robocerebra-artifacts/videos/
@@ -629,7 +629,7 @@ jobs:

      - name: Upload RoboCerebra eval metrics
        if: always()
-        uses: actions/upload-artifact@v7 # zizmor: ignore[unpinned-uses]
+        uses: actions/upload-artifact@v4 # zizmor: ignore[unpinned-uses]
        with:
          name: robocerebra-metrics
          path: /tmp/robocerebra-artifacts/metrics.json
@@ -648,19 +648,19 @@ jobs:
      ROBOMME_TASKS: PickXtimes,BinFill,StopCube,MoveCube,InsertPeg,SwingXtimes,VideoUnmask,ButtonUnmask,PickHighlight,PatternLock

    steps:
-      - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
+      - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2
        with:
          persist-credentials: false
          lfs: true

      - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v4 # zizmor: ignore[unpinned-uses]
+        uses: docker/setup-buildx-action@v3 # zizmor: ignore[unpinned-uses]
        with:
          cache-binary: false

      - name: Login to Docker Hub
        if: ${{ env.DOCKERHUB_USERNAME != '' }}
-        uses: docker/login-action@v4.4.0 # zizmor: ignore[unpinned-uses]
+        uses: docker/login-action@v3 # zizmor: ignore[unpinned-uses]
        with:
          username: ${{ secrets.DOCKERHUB_LEROBOT_USERNAME }}
          password: ${{ secrets.DOCKERHUB_LEROBOT_PASSWORD }}
@@ -668,7 +668,7 @@ jobs:
          DOCKERHUB_USERNAME: ${{ secrets.DOCKERHUB_LEROBOT_USERNAME }}

      - name: Build RoboMME benchmark image
-        uses: docker/build-push-action@v7 # zizmor: ignore[unpinned-uses]
+        uses: docker/build-push-action@v6 # zizmor: ignore[unpinned-uses]
        with:
          context: .
          file: docker/Dockerfile.benchmark.robomme
@@ -726,7 +726,7 @@ jobs:

      - name: Upload RoboMME rollout video
        if: always()
-        uses: actions/upload-artifact@v7 # zizmor: ignore[unpinned-uses]
+        uses: actions/upload-artifact@v4 # zizmor: ignore[unpinned-uses]
        with:
          name: robomme-rollout-video
          path: /tmp/robomme-artifacts/videos/
@@ -734,7 +734,7 @@ jobs:

      - name: Upload RoboMME eval metrics
        if: always()
-        uses: actions/upload-artifact@v7 # zizmor: ignore[unpinned-uses]
+        uses: actions/upload-artifact@v4 # zizmor: ignore[unpinned-uses]
        with:
          name: robomme-metrics
          path: /tmp/robomme-artifacts/metrics.json
@@ -754,19 +754,19 @@ jobs:
      LIBERO_PLUS_TASK_IDS: "[0,100,260,500,1000,1500,2000,2400]"

    steps:
-      - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1  # v7.0.1
+      - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
        with:
          persist-credentials: false
          lfs: true

      - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v4 # zizmor: ignore[unpinned-uses]
+        uses: docker/setup-buildx-action@v3 # zizmor: ignore[unpinned-uses]
        with:
          cache-binary: false

      - name: Login to Docker Hub
        if: ${{ env.DOCKERHUB_USERNAME != '' }}
-        uses: docker/login-action@v4.4.0 # zizmor: ignore[unpinned-uses]
+        uses: docker/login-action@v3 # zizmor: ignore[unpinned-uses]
        with:
          username: ${{ secrets.DOCKERHUB_LEROBOT_USERNAME }}
          password: ${{ secrets.DOCKERHUB_LEROBOT_PASSWORD }}
@@ -774,7 +774,7 @@ jobs:
          DOCKERHUB_USERNAME: ${{ secrets.DOCKERHUB_LEROBOT_USERNAME }}

      - name: Build LIBERO-plus benchmark image
-        uses: docker/build-push-action@v7 # zizmor: ignore[unpinned-uses]
+        uses: docker/build-push-action@v6 # zizmor: ignore[unpinned-uses]
        with:
          context: .
          file: docker/Dockerfile.benchmark.libero_plus
@@ -834,7 +834,7 @@ jobs:

      - name: Upload LIBERO-plus rollout video
        if: always()
-        uses: actions/upload-artifact@v7 # zizmor: ignore[unpinned-uses]
+        uses: actions/upload-artifact@v4 # zizmor: ignore[unpinned-uses]
        with:
          name: libero-plus-rollout-video
          path: /tmp/libero-plus-artifacts/videos/
@@ -842,7 +842,7 @@ jobs:

      - name: Upload LIBERO-plus eval metrics
        if: always()
-        uses: actions/upload-artifact@v7 # zizmor: ignore[unpinned-uses]
+        uses: actions/upload-artifact@v4 # zizmor: ignore[unpinned-uses]
        with:
          name: libero-plus-metrics
          path: /tmp/libero-plus-artifacts/metrics.json
@@ -858,19 +858,19 @@ jobs:
      HF_USER_TOKEN: ${{ secrets.LEROBOT_HF_USER }}

    steps:
-      - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1  # v7.0.1
+      - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
        with:
          persist-credentials: false
          lfs: true

      - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v4 # zizmor: ignore[unpinned-uses]
+        uses: docker/setup-buildx-action@v3 # zizmor: ignore[unpinned-uses]
        with:
          cache-binary: false

      - name: Login to Docker Hub
        if: ${{ env.DOCKERHUB_USERNAME != '' }}
-        uses: docker/login-action@v4.4.0 # zizmor: ignore[unpinned-uses]
+        uses: docker/login-action@v3 # zizmor: ignore[unpinned-uses]
        with:
          username: ${{ secrets.DOCKERHUB_LEROBOT_USERNAME }}
          password: ${{ secrets.DOCKERHUB_LEROBOT_PASSWORD }}
@@ -878,7 +878,7 @@ jobs:
          DOCKERHUB_USERNAME: ${{ secrets.DOCKERHUB_LEROBOT_USERNAME }}

      - name: Build VLABench benchmark image
-        uses: docker/build-push-action@v7 # zizmor: ignore[unpinned-uses]
+        uses: docker/build-push-action@v6 # zizmor: ignore[unpinned-uses]
        with:
          context: .
          file: docker/Dockerfile.benchmark.vlabench
@@ -936,7 +936,7 @@ jobs:

      - name: Upload VLABench rollout video
        if: always()
-        uses: actions/upload-artifact@v7 # zizmor: ignore[unpinned-uses]
+        uses: actions/upload-artifact@v4 # zizmor: ignore[unpinned-uses]
        with:
          name: vlabench-rollout-video
          path: /tmp/vlabench-artifacts/videos/
@@ -944,7 +944,7 @@ jobs:

      - name: Upload VLABench eval metrics
        if: always()
-        uses: actions/upload-artifact@v7 # zizmor: ignore[unpinned-uses]
+        uses: actions/upload-artifact@v4 # zizmor: ignore[unpinned-uses]
        with:
          name: vlabench-metrics
          path: /tmp/vlabench-artifacts/metrics.json
@@ -47,7 +47,7 @@ jobs:
    timeout-minutes: 30
    steps:
      - name: Checkout code
-        uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1  # v7.0.1
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
        with:
          persist-credentials: false

@@ -52,21 +52,21 @@ jobs:
          sudo apt-get update
          sudo apt-get install git-lfs
          git lfs install
-      - uses: actions/checkout@v7
+      - uses: actions/checkout@v6
        with:
          lfs: true
          persist-credentials: false
      - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v4 # zizmor: ignore[unpinned-uses]
+        uses: docker/setup-buildx-action@v3 # zizmor: ignore[unpinned-uses]
        with:
          cache-binary: false
      - name: Login to Docker Hub
-        uses: docker/login-action@v4.4.0 # zizmor: ignore[unpinned-uses]
+        uses: docker/login-action@v3 # zizmor: ignore[unpinned-uses]
        with:
          username: ${{ secrets.DOCKERHUB_LEROBOT_USERNAME }}
          password: ${{ secrets.DOCKERHUB_LEROBOT_PASSWORD }}
      - name: Build and push Docker image CPU
-        uses: docker/build-push-action@v7 # zizmor: ignore[unpinned-uses]
+        uses: docker/build-push-action@v6 # zizmor: ignore[unpinned-uses]
        with:
          context: .
          file: ./docker/Dockerfile.user
@@ -87,21 +87,21 @@ jobs:
          sudo apt-get update
          sudo apt-get install git-lfs
          git lfs install
-      - uses: actions/checkout@v7
+      - uses: actions/checkout@v6
        with:
          lfs: true
          persist-credentials: false
      - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v4 # zizmor: ignore[unpinned-uses]
+        uses: docker/setup-buildx-action@v3 # zizmor: ignore[unpinned-uses]
        with:
          cache-binary: false
      - name: Login to Docker Hub
-        uses: docker/login-action@v4.4.0 # zizmor: ignore[unpinned-uses]
+        uses: docker/login-action@v3 # zizmor: ignore[unpinned-uses]
        with:
          username: ${{ secrets.DOCKERHUB_LEROBOT_USERNAME }}
          password: ${{ secrets.DOCKERHUB_LEROBOT_PASSWORD }}
      - name: Build and push Docker image GPU
-        uses: docker/build-push-action@v7 # zizmor: ignore[unpinned-uses]
+        uses: docker/build-push-action@v6 # zizmor: ignore[unpinned-uses]
        with:
          context: .
          file: ./docker/Dockerfile.internal
@@ -33,7 +33,7 @@ jobs:
      github.event.workflow_run.event == 'pull_request' &&
      github.event.workflow_run.conclusion == 'success' &&
      github.repository == 'huggingface/lerobot'
-    uses: huggingface/doc-builder/.github/workflows/upload_pr_documentation.yml@6108e850ae1cf2f71bb0815a600bcd50c39abfa7  # main
+    uses: huggingface/doc-builder/.github/workflows/upload_pr_documentation.yml@2430c1ec91d04667414e2fa31ecfc36c153ea391  # main
    with:
      package_name: lerobot
    secrets:
@@ -55,7 +55,7 @@ jobs:
      github.repository == 'huggingface/lerobot'
    permissions:
      contents: read
-    uses: huggingface/doc-builder/.github/workflows/build_main_documentation.yml@6108e850ae1cf2f71bb0815a600bcd50c39abfa7  # main
+    uses: huggingface/doc-builder/.github/workflows/build_main_documentation.yml@e60a538eea9817ab312196d0d233604b01697265  # main
    with:
      commit_sha: ${{ github.sha }}
      package: lerobot
@@ -78,7 +78,7 @@ jobs:
    permissions:
      contents: read
      pull-requests: write
-    uses: huggingface/doc-builder/.github/workflows/build_pr_documentation.yml@6108e850ae1cf2f71bb0815a600bcd50c39abfa7  # main
+    uses: huggingface/doc-builder/.github/workflows/build_pr_documentation.yml@e60a538eea9817ab312196d0d233604b01697265  # main
    with:
      commit_sha: ${{ github.event.pull_request.head.sha }}
      pr_number: ${{ github.event.number }}
@@ -69,7 +69,7 @@ jobs:
      HF_LEROBOT_HOME: /mnt/cache/.cache/huggingface/lerobot
      HF_USER_TOKEN: ${{ secrets.LEROBOT_HF_USER }}
    steps:
-      - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1  # v7.0.1
+      - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
        with:
          persist-credentials: false
          lfs: true
@@ -87,7 +87,7 @@ jobs:
          libusb-1.0-0-dev speech-dispatcher libgeos-dev portaudio19-dev

      - name: Setup uv and Python
-        uses: astral-sh/setup-uv@11f9893b081a58869d3b5fccaea48c9e9e46f990  # v8.3.2
+        uses: astral-sh/setup-uv@d0cc045d04ccac9d8b7881df0226f9e82c39688e  # v6
        with:
          enable-cache: true
          version: ${{ env.UV_VERSION }}
@@ -63,7 +63,7 @@ jobs:
      HF_LEROBOT_HOME: /mnt/cache/.cache/huggingface/lerobot
      HF_USER_TOKEN: ${{ secrets.LEROBOT_HF_USER }}
    steps:
-      - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1  # v7.0.1
+      - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
        with:
          lfs: true
          persist-credentials: false
@@ -80,7 +80,7 @@ jobs:
          speech-dispatcher libgeos-dev portaudio19-dev

      - name: Setup uv and Python
-        uses: astral-sh/setup-uv@11f9893b081a58869d3b5fccaea48c9e9e46f990  # v8.3.2
+        uses: astral-sh/setup-uv@d0cc045d04ccac9d8b7881df0226f9e82c39688e  # v6
        with:
          enable-cache: true
          version: ${{ env.UV_VERSION }}
@@ -137,21 +137,21 @@ jobs:
          sudo apt-get update
          sudo apt-get install git-lfs
          git lfs install
-      - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1  # v7.0.1
+      - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
        with:
          lfs: true
          persist-credentials: false
      - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@bb05f3f5519dd87d3ba754cc423b652a5edd6d2c  # v4.2.0
+        uses: docker/setup-buildx-action@8d2750c68a42422c14e847fe6c8ac0403b4cbd6f  # v3
        with:
          cache-binary: false
      - name: Login to Docker Hub
-        uses: docker/login-action@af1e73f918a031802d376d3c8bbc3fe56130a9b0  # v4.4.0
+        uses: docker/login-action@c94ce9fb468520275223c153574b00df6fe4bcc9  # v3
        with:
          username: ${{ secrets.DOCKERHUB_LEROBOT_USERNAME }}
          password: ${{ secrets.DOCKERHUB_LEROBOT_PASSWORD }}
      - name: Build and push Docker image
-        uses: docker/build-push-action@53b7df96c91f9c12dcc8a07bcb9ccacbed38856a  # v7.3.0
+        uses: docker/build-push-action@10e90e3645eae34f1e60eeb005ba3a3d33f178e8  # v6
        with:
          context: .
          file: ./docker/Dockerfile.internal
@@ -29,7 +29,7 @@ jobs:
    runs-on: ubuntu-latest
    if: github.repository == 'huggingface/lerobot'
    steps:
-      - uses: actions/github-script@v9
+      - uses: actions/github-script@v8
        with:
          script: |
            // Setup Input Text
@@ -48,12 +48,12 @@ jobs:
    outputs:
      changed: ${{ steps.diff.outputs.changed }}
    steps:
-      - uses: actions/checkout@v7
+      - uses: actions/checkout@v6
        with:
          persist-credentials: false

      - name: Setup uv and Python
-        uses: astral-sh/setup-uv@v8.3.2 # zizmor: ignore[unpinned-uses]
+        uses: astral-sh/setup-uv@v6 # zizmor: ignore[unpinned-uses]
        with:
          version: ${{ env.UV_VERSION }}
          python-version: ${{ env.PYTHON_VERSION }}
@@ -74,7 +74,7 @@ jobs:

      - name: Upload updated lockfile
        if: steps.diff.outputs.changed == 'true'
-        uses: actions/upload-artifact@v7 # zizmor: ignore[unpinned-uses]
+        uses: actions/upload-artifact@v4 # zizmor: ignore[unpinned-uses]
        with:
          name: uv-lock
          path: uv.lock
@@ -93,13 +93,13 @@ jobs:
      HF_LEROBOT_HOME: /mnt/cache/.cache/huggingface/lerobot
      HF_USER_TOKEN: ${{ secrets.LEROBOT_HF_USER }}
    steps:
-      - uses: actions/checkout@v7
+      - uses: actions/checkout@v6
        with:
          lfs: true
          persist-credentials: false

      - name: Download updated lockfile
-        uses: actions/download-artifact@v8 # zizmor: ignore[unpinned-uses]
+        uses: actions/download-artifact@v4 # zizmor: ignore[unpinned-uses]
        with:
          name: uv-lock

@@ -115,7 +115,7 @@ jobs:
          speech-dispatcher libgeos-dev portaudio19-dev

      - name: Setup uv and Python
-        uses: astral-sh/setup-uv@v8.3.2 # zizmor: ignore[unpinned-uses]
+        uses: astral-sh/setup-uv@v6 # zizmor: ignore[unpinned-uses]
        with:
          enable-cache: true
          version: ${{ env.UV_VERSION }}
@@ -153,27 +153,27 @@ jobs:
          sudo apt-get update
          sudo apt-get install git-lfs
          git lfs install
-      - uses: actions/checkout@v7
+      - uses: actions/checkout@v6
        with:
          lfs: true
          persist-credentials: false

      - name: Download updated lockfile
-        uses: actions/download-artifact@v8 # zizmor: ignore[unpinned-uses]
+        uses: actions/download-artifact@v4 # zizmor: ignore[unpinned-uses]
        with:
          name: uv-lock

      - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v4 # zizmor: ignore[unpinned-uses]
+        uses: docker/setup-buildx-action@v3 # zizmor: ignore[unpinned-uses]
        with:
          cache-binary: false
      - name: Login to Docker Hub
-        uses: docker/login-action@v4.4.0 # zizmor: ignore[unpinned-uses]
+        uses: docker/login-action@v3 # zizmor: ignore[unpinned-uses]
        with:
          username: ${{ secrets.DOCKERHUB_LEROBOT_USERNAME }}
          password: ${{ secrets.DOCKERHUB_LEROBOT_PASSWORD }}
      - name: Build and push Docker image
-        uses: docker/build-push-action@v7 # zizmor: ignore[unpinned-uses]
+        uses: docker/build-push-action@v6 # zizmor: ignore[unpinned-uses]
        with:
          context: .
          file: ./docker/Dockerfile.internal
@@ -247,12 +247,12 @@ jobs:
    env:
      GH_TOKEN: ${{ secrets.UPDATE_LOCK_TOKEN }}
    steps:
-      - uses: actions/checkout@v7
+      - uses: actions/checkout@v6
        with:
          persist-credentials: false

      - name: Download updated lockfile
-        uses: actions/download-artifact@v8 # zizmor: ignore[unpinned-uses]
+        uses: actions/download-artifact@v4 # zizmor: ignore[unpinned-uses]
        with:
          name: uv-lock

@@ -33,7 +33,7 @@ jobs:
    runs-on: ubuntu-latest
    if: github.repository == 'huggingface/lerobot' && !github.event.pull_request.draft
    steps:
-      - uses: actions/labeler@v7
+      - uses: actions/labeler@v6
        with:
          repo-token: ${{ secrets.GITHUB_TOKEN }}
          sync-labels: true # Removes labels if files are removed from the PR
@@ -43,12 +43,12 @@ jobs:
    runs-on: ubuntu-latest
    steps:
      - name: Checkout code
-        uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1  # v7.0.1
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
        with:
          persist-credentials: false

      - name: Set up Python
-        uses: actions/setup-python@5fda3b95a4ea91299a34e894583c3862153e4b97  # v6
+        uses: actions/setup-python@a309ff8b426b58ec0e2a45f0f869d46889d02405  # v6
        with:
          python-version: '3.12'

@@ -38,12 +38,12 @@ jobs:

    steps:
      - name: Checkout code
-        uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1  # v7.0.1
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
        with:
          persist-credentials: false

      - name: Set up Python
-        uses: actions/setup-python@5fda3b95a4ea91299a34e894583c3862153e4b97  # v6
+        uses: actions/setup-python@a309ff8b426b58ec0e2a45f0f869d46889d02405  # v6
        with:
          python-version: '3.12'

@@ -104,7 +104,7 @@ jobs:
      - name: Publish to TestPyPI for pre-releases
        # True for tags like 'v0.2.0-rc1'
        if: startsWith(github.ref, 'refs/tags/v') && contains(github.ref, '-')
-        uses: pypa/gh-action-pypi-publish@ba38be9e461d3875417946c167d0b5f3d385a247  # v1.14.1
+        uses: pypa/gh-action-pypi-publish@ed0c53931b1dc9bd32cbe73a98c7f6766f8a527e  # v1.13.0
        with:
          repository-url: https://test.pypi.org/legacy/
          verbose: true
@@ -112,7 +112,7 @@ jobs:

      - name: Publish to PyPI
        if: startsWith(github.ref, 'refs/tags/v') && !contains(github.ref, '-')
-        uses: pypa/gh-action-pypi-publish@ba38be9e461d3875417946c167d0b5f3d385a247  # v1.14.1
+        uses: pypa/gh-action-pypi-publish@ed0c53931b1dc9bd32cbe73a98c7f6766f8a527e  # v1.13.0
        with:
          verbose: true
          print-hash: true
@@ -127,7 +127,7 @@ jobs:
    env:
      MUJOCO_GL: egl
    steps:
-      - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1  # v7.0.1
+      - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
        with:
          lfs: true
          persist-credentials: false
@@ -137,7 +137,7 @@ jobs:
          git curl libglib2.0-0 libegl1-mesa-dev ffmpeg libusb-1.0-0-dev \
          speech-dispatcher libgeos-dev portaudio19-dev
      - name: Setup uv and Python
-        uses: astral-sh/setup-uv@11f9893b081a58869d3b5fccaea48c9e9e46f990  # v8.3.2
+        uses: astral-sh/setup-uv@d0cc045d04ccac9d8b7881df0226f9e82c39688e  # v6
        with:
          enable-cache: true # zizmor: ignore[cache-poisoning]
          version: ${{ env.UV_VERSION }}
@@ -43,12 +43,12 @@ jobs:
    runs-on: ubuntu-latest
    steps:
      - name: Checkout code
-        uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1  # v7.0.1
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
        with:
          fetch-depth: 0
          persist-credentials: false

      - name: Secret Scanning
-        uses: trufflesecurity/trufflehog@27b0417c16317ca9a472a9a8092acce143b49c55  # v3.95.9
+        uses: trufflesecurity/trufflehog@eafb8c5f6a06175141c27f17bcc17941853d0047  # v3.90.0
        with:
          extra_args: --only-verified
@@ -519,6 +519,13 @@ def compute_episode_stats(
        if features[key]["dtype"] in {"string", "language"}:
            continue

+        # Features with a zero-width dimension contain no statistics-bearing
+        # values. Skip them like strings instead of letting
+        # get_feature_stats -> RunningQuantileStats.update reshape a size-0 array,
+        # which raises "ValueError: cannot reshape array of size 0".
+        if any(dim == 0 for dim in features[key].get("shape", ())):
+            continue
+
        if features[key]["dtype"] in ["image", "video"]:
            ep_ft_array = sample_images(data)
            axes_to_reduce = (0, 2, 3)
@@ -172,6 +172,23 @@ class DatasetWriter:
    def _get_image_file_dir(self, episode_index: int, image_key: str) -> Path:
        return self._get_image_file_path(episode_index, image_key, frame_index=0).parent

+    def _get_episode_buffer_index(self) -> int:
+        episode_index = self.episode_buffer["episode_index"]
+        # episode_index is `int` when freshly created, but becomes `np.ndarray` after
+        # save_episode() mutates the buffer. Handle both types here.
+        if isinstance(episode_index, np.ndarray):
+            episode_index = episode_index.item() if episode_index.size == 1 else episode_index[0]
+        return int(episode_index)
+
+    def _delete_camera_frame_dirs(self, camera_keys: list[str]) -> None:
+        if self.image_writer is not None:
+            self._wait_image_writer()
+        episode_index = self._get_episode_buffer_index()
+        for camera_key in camera_keys:
+            img_dir = self._get_image_file_dir(episode_index, camera_key)
+            if img_dir.is_dir():
+                shutil.rmtree(img_dir)
+
    def _save_image(
        self, image: torch.Tensor | np.ndarray | PIL.Image.Image, fpath: Path, compress_level: int = 1
    ) -> None:
@@ -369,7 +386,9 @@ class DatasetWriter:
                self._episodes_since_last_encoding = 0

        if episode_data is None:
-            self.clear_episode_buffer(delete_images=len(self._meta.image_keys) > 0)
+            if len(self._meta.image_keys) > 0:
+                self._delete_camera_frame_dirs(self._meta.image_keys)
+            self.episode_buffer = self._create_episode_buffer()

    def _batch_save_episode_video(self, start_episode: int, end_episode: int | None = None) -> None:
        """Batch save videos for multiple episodes."""
@@ -561,10 +580,10 @@ class DatasetWriter:
        return metadata

    def clear_episode_buffer(self, delete_images: bool = True) -> None:
-        """Discard the current episode buffer and optionally delete temp images.
+        """Discard the current episode buffer and optionally delete temp camera frames.

        Args:
-            delete_images: If ``True``, remove temporary image directories
+            delete_images: If ``True``, remove temporary camera frame directories
                written for the current episode.
        """
        # Cancel streaming encoder if active
@@ -572,17 +591,7 @@ class DatasetWriter:
            self._streaming_encoder.cancel_episode()

        if delete_images:
-            if self.image_writer is not None:
-                self._wait_image_writer()
-            episode_index = self.episode_buffer["episode_index"]
-            # episode_index is `int` when freshly created, but becomes `np.ndarray` after
-            # save_episode() mutates the buffer. Handle both types here.
-            if isinstance(episode_index, np.ndarray):
-                episode_index = episode_index.item() if episode_index.size == 1 else episode_index[0]
-            for cam_key in self._meta.image_keys:
-                img_dir = self._get_image_file_dir(episode_index, cam_key)
-                if img_dir.is_dir():
-                    shutil.rmtree(img_dir)
+            self._delete_camera_frame_dirs(self._meta.camera_keys)

        self.episode_buffer = self._create_episode_buffer()

@@ -64,12 +64,20 @@ def get_hf_features_from_features(features: dict) -> datasets.Features:
            continue
        elif ft["dtype"] == "image":
            hf_features[key] = datasets.Image()
+        elif len(ft["shape"]) > 1 and any(dim == 0 for dim in ft["shape"]):
+            raise ValueError(
+                f"Multidimensional features with a zero-width dimension are not supported: "
+                f"'{key}' has shape {ft['shape']}. Only the one-dimensional shape (0,) is supported."
+            )
        elif ft["shape"] == (1,):
            hf_features[key] = datasets.Value(dtype=ft["dtype"])
        elif len(ft["shape"]) == 1:
-            hf_features[key] = datasets.Sequence(
-                length=ft["shape"][0], feature=datasets.Value(dtype=ft["dtype"])
-            )
+            # A zero-width feature (shape=(0,)) has no fixed-size Arrow representation:
+            # pyarrow rejects a fixed-size list of length 0 ("list_size needs to be a
+            # strict positive integer"). Store it as a variable-length sequence
+            # (length=-1) so each per-frame value is simply an empty list.
+            seq_length = ft["shape"][0] if ft["shape"][0] > 0 else -1
+            hf_features[key] = datasets.Sequence(length=seq_length, feature=datasets.Value(dtype=ft["dtype"]))
        elif len(ft["shape"]) == 2:
            hf_features[key] = datasets.Array2D(shape=ft["shape"], dtype=ft["dtype"])
        elif len(ft["shape"]) == 3:
@@ -453,6 +453,9 @@ def eval_policy(
            raise exc from None

    start = time.time()
+    # Preserve the mode for direct callers. eval_policy_all scopes the mode
+    # around all tasks so parallel evaluations cannot race with each other.
+    was_training = policy.training
    policy.eval()

    # Determine how many batched rollouts we need to get n_episodes. Note that if n_episodes is not evenly
@@ -674,6 +677,8 @@ def eval_policy(
    if save_predicted_video:
        info["predicted_video_paths"] = predicted_video_paths

+    policy.train(was_training)
+
    return info


@@ -1010,40 +1015,48 @@ def eval_policy_all(
        recording_private=recording_private,
    )

-    if max_parallel_tasks <= 1:
-        prefetch_thread: threading.Thread | None = None
-        for i, (task_group, task_id, env) in enumerate(tasks):
-            if prefetch_thread is not None:
-                prefetch_thread.join()
-                prefetch_thread = None
+    # Set the shared policy's mode before launching any workers. Restoring it
+    # inside individual tasks would let one task enable training mode while
+    # another task is still evaluating.
+    was_training = policy.training
+    policy.eval()
+    try:
+        if max_parallel_tasks <= 1:
+            prefetch_thread: threading.Thread | None = None
+            for i, (task_group, task_id, env) in enumerate(tasks):
+                if prefetch_thread is not None:
+                    prefetch_thread.join()
+                    prefetch_thread = None

-            try:
-                tg, tid, metrics = task_runner(task_group, task_id, env)
-                _accumulate_to(tg, metrics)
-                per_task_infos.append({"task_group": tg, "task_id": tid, "metrics": metrics})
-            finally:
-                env.close()
-                # Prefetch next task's workers *after* closing current env to prevent
-                # GPU memory overlap between consecutive tasks.
-                if i + 1 < len(tasks):
-                    next_env = tasks[i + 1][2]
-                    if hasattr(next_env, "_ensure"):
-                        prefetch_thread = threading.Thread(target=next_env._ensure, daemon=True)
-                        prefetch_thread.start()
-    else:
-        with cf.ThreadPoolExecutor(max_workers=max_parallel_tasks) as executor:
-            fut2meta = {}
-            for task_group, task_id, env in tasks:
-                fut = executor.submit(task_runner, task_group, task_id, env)
-                fut2meta[fut] = (task_group, task_id, env)
-            for fut in cf.as_completed(fut2meta):
-                tg, tid, env = fut2meta[fut]
                try:
-                    tg, tid, metrics = fut.result()
+                    tg, tid, metrics = task_runner(task_group, task_id, env)
                    _accumulate_to(tg, metrics)
                    per_task_infos.append({"task_group": tg, "task_id": tid, "metrics": metrics})
                finally:
                    env.close()
+                    # Prefetch next task's workers *after* closing current env to prevent
+                    # GPU memory overlap between consecutive tasks.
+                    if i + 1 < len(tasks):
+                        next_env = tasks[i + 1][2]
+                        if hasattr(next_env, "_ensure"):
+                            prefetch_thread = threading.Thread(target=next_env._ensure, daemon=True)
+                            prefetch_thread.start()
+        else:
+            with cf.ThreadPoolExecutor(max_workers=max_parallel_tasks) as executor:
+                fut2meta = {}
+                for task_group, task_id, env in tasks:
+                    fut = executor.submit(task_runner, task_group, task_id, env)
+                    fut2meta[fut] = (task_group, task_id, env)
+                for fut in cf.as_completed(fut2meta):
+                    tg, tid, env = fut2meta[fut]
+                    try:
+                        tg, tid, metrics = fut.result()
+                        _accumulate_to(tg, metrics)
+                        per_task_infos.append({"task_group": tg, "task_id": tid, "metrics": metrics})
+                    finally:
+                        env.close()
+    finally:
+        policy.train(was_training)

    # compute aggregated metrics helper (robust to lists/scalars)
    def _agg_from_list(xs):
@@ -453,9 +453,11 @@ def record(
                encoder_queue_maxsize=cfg.dataset.encoder_queue_maxsize,
            )

-        robot.connect()
+        # Connect the teleoperator before the robot so the robot isn't left idle (and possibly
+        # tripping a firmware watchdog) during teleop init. Matches lerobot_teleoperate.py.
        if teleop is not None:
            teleop.connect()
+        robot.connect()

        listener, events = init_keyboard_listener()

@@ -687,6 +687,26 @@ def test_compute_episode_stats_string_features_skipped():
    assert "q01" in stats["action"]


+@pytest.mark.parametrize("shape", [(0,), (0, 2), (2, 0), (1, 0, 2)])
+def test_compute_episode_stats_zero_width_feature_skipped(shape):
+    """Features with any zero-width dimension carry no values and are skipped."""
+    episode_data = {
+        "action": np.random.normal(0, 1, (100, 5)).astype(np.float32),
+        "target": np.zeros((100, *shape), dtype=np.float32),
+    }
+    features = {
+        "action": {"dtype": "float32", "shape": (5,)},
+        "target": {"dtype": "float32", "shape": shape},
+    }
+
+    stats = compute_episode_stats(episode_data, features)
+
+    # Zero-width features are skipped, just like strings; non-empty features are unaffected.
+    assert "target" not in stats
+    assert "action" in stats
+    assert "q01" in stats["action"]
+
+
 def test_aggregate_feature_stats_with_quantiles():
    """Test aggregating feature stats that include quantiles."""
    stats_ft_list = [
@@ -27,6 +27,7 @@ pytest.importorskip("datasets", reason="datasets is required (install lerobot[da

 from lerobot.configs import VideoEncoderConfig
 from lerobot.datasets.dataset_writer import _encode_video_worker
+from lerobot.datasets.feature_utils import get_hf_features_from_features
 from lerobot.datasets.lerobot_dataset import LeRobotDataset
 from lerobot.datasets.utils import DEFAULT_IMAGE_PATH
 from tests.fixtures.constants import DEFAULT_FPS, DUMMY_REPO_ID
@@ -189,6 +190,36 @@ def test_save_multiple_episodes(tmp_path):
    assert dataset.meta.total_frames == total_frames


+def test_save_episode_with_zero_width_feature(tmp_path):
+    """A one-dimensional empty numeric feature round-trips and has no statistics."""
+    features = {
+        **SIMPLE_FEATURES,
+        "target": {"dtype": "float32", "shape": (0,), "names": None},
+    }
+    root = tmp_path / "ds"
+    dataset = LeRobotDataset.create(repo_id=DUMMY_REPO_ID, fps=DEFAULT_FPS, features=features, root=root)
+    for _ in range(4):
+        dataset.add_frame(_make_frame(features))
+    dataset.save_episode()
+    dataset.finalize()
+
+    assert dataset.meta.total_episodes == 1
+    assert dataset.meta.total_frames == 4
+
+    reloaded = LeRobotDataset(repo_id=DUMMY_REPO_ID, root=root)
+    target = np.asarray(reloaded[0]["target"])
+    assert target.shape == (0,)
+    assert "target" not in (reloaded.meta.stats or {})
+
+
+@pytest.mark.parametrize("shape", [(0, 2), (2, 0), (1, 0, 2)])
+def test_multidimensional_zero_width_feature_rejected(shape):
+    features = {"target": {"dtype": "float32", "shape": shape, "names": None}}
+
+    with pytest.raises(ValueError, match="Multidimensional features with a zero-width dimension"):
+        get_hf_features_from_features(features)
+
+
 # ── clear / lifecycle ────────────────────────────────────────────────


@@ -204,6 +235,38 @@ def test_clear_resets_buffer(tmp_path):
    assert dataset.writer.episode_buffer["size"] == 0


+def test_clear_removes_video_frame_staging_dir(tmp_path):
+    """clear_episode_buffer() removes PNG staging dirs for video features."""
+    video_key = "observation.images.cam"
+    features = {
+        video_key: {
+            "dtype": "video",
+            "shape": (64, 96, 3),
+            "names": ["height", "width", "channels"],
+        },
+        "action": {"dtype": "float32", "shape": (2,), "names": None},
+    }
+    dataset = LeRobotDataset.create(
+        repo_id=DUMMY_REPO_ID,
+        fps=DEFAULT_FPS,
+        features=features,
+        root=tmp_path / "ds",
+        use_videos=True,
+    )
+
+    dataset.add_frame(_make_frame(features))
+    video_staging_dir = (
+        dataset.root
+        / Path(DEFAULT_IMAGE_PATH.format(image_key=video_key, episode_index=0, frame_index=0)).parent
+    )
+    assert video_staging_dir.is_dir()
+
+    dataset.clear_episode_buffer()
+
+    assert dataset.writer.episode_buffer["size"] == 0
+    assert not video_staging_dir.exists()
+
+
 def test_finalize_is_idempotent(tmp_path):
    """Calling finalize() twice does not raise."""
    dataset = LeRobotDataset.create(
Author	SHA1	Message	Date
Steven Palma	45243dcf7c	chore(dataset): add check dataset shape	2026-07-27 18:36:15 +02:00
Steven Palma	034693f724	Merge branch 'main' into fix/save-episode-zero-width-feature	2026-07-27 15:52:12 +02:00
Steven Palma	bbeacfe57d	fix(record): connect teleoperator before robot to avoid watchdog jump (#4166 ) * fix(record): connect teleoperator before robot to avoid watchdog jump lerobot-record connected the robot before the teleoperator. A robot's connect()/reset() can leave it holding a default pose under a firmware watchdog (e.g. Unitree G1); if teleop.connect() (model loading, IK init, network setup) then takes longer than that watchdog, the joints drop to damping and the first send_action() makes the robot jump. Swap the order so the teleoperator connects first, matching the ordering already used in lerobot_teleoperate.py. Pure ordering fix, no API change. Fixes #3684 * fix(record): trim comment and add connect-order regression test Address review feedback on #3684: - Trim the verbose ordering comment down to two lines. - Add test_record_connects_teleop_before_robot to tests/test_control_robot.py, asserting teleop.connect() runs before robot.connect() in record(). * chore(test): remove test --------- Co-authored-by: Jaimin Patel <jpatel@tuvalabs.com> Co-authored-by: Martino Russi <77496684+nepyope@users.noreply.github.com>	2026-07-27 14:09:58 +02:00
Steven Palma	801346e18c	fix(scripts): restore policy training mode after eval_policy() in lerobot-eval (#4162 ) * fix(scripts/eval): restore policy training mode after eval_policy() `eval_policy` calls `policy.eval()` before the rollout but never restores the prior mode on return. When called from the training loop (`lerobot_train.py`'s `eval_policy_all -> run_one -> eval_one -> eval_policy` chain), the policy is left in eval mode for every subsequent training step, which silently: * disables Dropout (no regularisation), * freezes BatchNorm running stats (no further EMA updates). Under DDP only `is_main_process` runs eval (lerobot_train.py:527), so the main rank ends up in eval mode while workers stay in train mode — the all-reduced gradients then combine forward passes computed with different dropout masks and different BN behaviour, a real DDP-correctness issue. Scope of impact: * Affects every policy with Dropout in its forward path. In-tree, that includes the default ACT (6 Dropout layers at p=0.1), Diffusion (vision backbone), VQ-BeT, Multi-Task DiT, X-VLA, plus all VLA policies that inherit Dropout from their pretrained HF backbone (PI0/PI0.5/PI0-FAST, SmolVLA, GR00T-N1.5, EO1, Wall-X). * Triggers from the first eval onward. On the default config (steps=100k, eval_freq=20k) that's 80% of training; on the LIBERO / RoboCasa / VLABench example commands in docs/ (eval_freq=1k–5k) it's 95–99% of training. * Policies using only LayerNorm/GroupNorm and no Dropout (TDMPC, RTC) are unaffected. Policies using `FrozenBatchNorm2d` (ACT's ResNet backbone) are immune to the BN-stat half; the Dropout half still bites. Fix: * Snapshot `policy.training` on entry to `eval_policy`. * Restore it on normal return. * Save-and-restore is a strict no-op for callers that pass an already-eval-mode policy (e.g. the standalone `lerobot-eval` script loading a frozen checkpoint). * Restoration is placed before the normal return only, not in a try/finally — exception paths leave the policy in eval mode, same as today. A try/finally upgrade would require re-indenting ~165 lines and can land as a separate cleanup if desired. Tests (tests/scripts/test_eval.py, 7 tests total, ~1.6s): * Regression gates on the lerobot_eval fix itself: training-mode preservation, eval-mode preservation, dropout-active behavioural check, non-crash for both entry modes. * Quantitative mechanism demonstration (`test_missing_mode_restoration_hurts_generalisation`): trains a tiny Dropout+BatchNorm MLP under both the bug pattern and the fix pattern on identical data and seed, then asserts the buggy variant generalises at least 5% worse on a held-out val set. In repeated runs we see 10-25% deltas on this toy problem; real policies (more layers, more Dropout, longer training) generally see larger gaps. Lives alongside the regression tests so the empirical proof is reproducible from the repo without adding a separate benchmarks/ directory. * fix(scripts): keep policy train/eval --------- Co-authored-by: ModeEric <ericjm4@illinois.edu>	2026-07-27 14:08:11 +02:00
MihaiAnca13	ab87fd9764	fix(datasets): clear video frame staging on episode reset (#3683 ) * fix video frame staging cleanup on episode reset * linting --------- Co-authored-by: Steven Palma <imstevenpmwork@ieee.org>	2026-07-27 13:44:32 +02:00
pranjalthebhatia	bb3ef3537f	fix(datasets): support features with a zero-width dimension (shape=(0,)) Declaring a numeric feature with `shape=(0,)` crashed `save_episode()` in two distinct places, leaving `dtype: "string"` as the only (type-lossy) workaround: - `compute_episode_stats` -> `RunningQuantileStats.update` reshaped a size-0 array, raising "ValueError: cannot reshape array of size 0 into shape (0)". - `get_hf_features_from_features` mapped it to a fixed-size Arrow list of length 0, which pyarrow rejects ("list_size needs to be a strict positive integer"). The issue only reported the first error; the second surfaces once the first is fixed. This change handles both: - Skip zero-width features during episode stats, exactly as string/language features are already skipped. - Store 1-D zero-width features as a variable-length sequence (length=-1) so each per-frame value is simply an empty list. Adds a unit test (stats layer) and an integration test that records, saves, and reads back a zero-width feature, asserting it round-trips as an empty vector and is excluded from stats. Fixes #3654 Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>	2026-06-04 17:42:55 -04:00