From 9a6ca751bb26c37508b3c330dc6cbf6432624338 Mon Sep 17 00:00:00 2001 From: Claudio Coppola Date: Mon, 27 Jan 2025 17:26:47 +0000 Subject: [PATCH] fix: aloha_hd5 to LerobotDataset v2 frame appending out of the right scope --- examples/port_datasets/aloha_hd5.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/port_datasets/aloha_hd5.py b/examples/port_datasets/aloha_hd5.py index 2a6bb590..91c8c1cd 100644 --- 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