adds more annotations

lerobot/common/logger.py

@@ -241,5 +241,6 @@ class Logger:
 
     def log_video(self, video_path: str, step: int, mode: str = "train"):
         assert mode in {"train", "eval"}
+        assert self._wandb is not None
         wandb_video = self._wandb.Video(video_path, fps=self._cfg.fps, format="mp4")
         self._wandb.log({f"{mode}/video": wandb_video}, step=step)

lerobot/scripts/push_dataset_to_hub.py

@@ -97,7 +97,9 @@ def get_from_raw_to_lerobot_format_fn(raw_format: str):
     return from_raw_to_lerobot_format
 
 
-def save_meta_data(info: dict[str, Any], stats: dict, episode_data_index, meta_data_dir: Path):
+def save_meta_data(
+    info: dict[str, Any], stats: dict, episode_data_index: dict[str, list], meta_data_dir: Path
+):
     meta_data_dir.mkdir(parents=True, exist_ok=True)
 
     # save info