wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix consolidate

This commit is contained in:
Simon Alibert 2024-10-23 18:45:59 +02:00
parent 07570f867f
commit 1aba80d93f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/common/datasets/lerobot_dataset.py
View File

@ -704,7 +704,7 @@ class LeRobotDataset(torch.utils.data.Dataset):
if len(self.video_keys) > 0: if len(self.video_keys) > 0:
self.encode_videos() 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) shutil.rmtree(self.image_writer.dir)
if run_compute_stats: if run_compute_stats: