nit

huggingface · Dec 1, 2024 · ad2cdaf · ad2cdaf
1 parent 69609ad
commit ad2cdaf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lerobot/common/datasets/push_dataset_to_hub/openx_rlds_format.py b/lerobot/common/datasets/push_dataset_to_hub/openx_rlds_format.py
@@ -208,12 +208,12 @@ def load_from_raw(
         ep_dict["episode_index"] = torch.tensor([ep_idx] * num_frames)
         ep_dict["frame_index"] = torch.arange(0, num_frames, 1)
 
+        image_array_dict = {key: [] for key in image_keys}
+
         for im_key in image_keys:
             imgs = episode["observation"][im_key]
             image_array_dict[im_key] = [tf_img_convert(img) for img in imgs]
 
-        image_array_dict = {key: [] for key in image_keys}
-
         # loop through all cameras
         for im_key in image_keys:
             img_key = f"observation.images.{im_key}"