From c2388c59bed426cf4cd17ab3496fc9ea4cf3a7bd Mon Sep 17 00:00:00 2001 From: Remi Cadene Date: Wed, 10 Jul 2024 19:07:15 +0200 Subject: [PATCH] uncomment shutil.rmtree(tmp_imgs_dir) --- lerobot/scripts/control_robot.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lerobot/scripts/control_robot.py b/lerobot/scripts/control_robot.py index 137e8b78..f87e14d9 100644 --- a/lerobot/scripts/control_robot.py +++ b/lerobot/scripts/control_robot.py @@ -287,11 +287,11 @@ def record_dataset( tmp_imgs_dir = videos_dir / f"{key}_episode_{episode_index:06d}" fname = f"{key}_episode_{episode_index:06d}.mp4" video_path = local_dir / "videos" / fname + # note: video encoding is not done asynchronously encode_video_frames(tmp_imgs_dir, video_path, fps) - # TODO(rcadene): uncomment? # clean temporary images directory - # shutil.rmtree(tmp_imgs_dir) + shutil.rmtree(tmp_imgs_dir) # store the reference to the video frame ep_dict[key] = []