Improves Type Annotations (#252)

2024-06-10 19:09:48 +01:00
parent a06598678c
commit 54c9776bde
7 changed files with 54 additions and 23 deletions
--- a/lerobot/common/logger.py
+++ b/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)