Fix consolidate

2024-10-23 18:45:59 +02:00 · 2024-10-23 18:45:59 +02:00 · 1aba80d93f
parent 07570f867f
commit 1aba80d93f
1 changed files with 1 additions and 1 deletions
--- a/lerobot/common/datasets/lerobot_dataset.py
+++ b/lerobot/common/datasets/lerobot_dataset.py
@ -704,7 +704,7 @@ class LeRobotDataset(torch.utils.data.Dataset):
        if len(self.video_keys) > 0:
            self.encode_videos()

-        if not keep_image_files:
+        if not keep_image_files and self.image_writer is not None:
            shutil.rmtree(self.image_writer.dir)

        if run_compute_stats: