fix: aloha_hd5 to LerobotDataset v2 frame appending out of the right scope

2025-01-27 17:26:47 +00:00 · 2025-01-27 17:26:47 +00:00 · 9a6ca751bb
parent d8e4a2ccd7
commit 9a6ca751bb
1 changed files with 1 additions and 1 deletions
--- a/examples/port_datasets/aloha_hd5.py
+++ b/examples/port_datasets/aloha_hd5.py
@ -132,7 +132,7 @@ class AlohaHD5Extractor:
                        frame[feature_id] = torch.from_numpy(image.transpose(2, 0, 1))
                    else:
                        frame[feature_id] = torch.from_numpy(file[feature_name_hd5][frame_idx])
-            frames.append(frame)
+                frames.append(frame)
        return frames

    @staticmethod