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

Minor improvements in push_dataset_to_hub

This commit is contained in:
Remi Cadene 2024-05-17 14:50:55 +00:00
parent c6fe57ede6
commit 729e2e550b
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
lerobot/scripts/push_dataset_to_hub.py
View File

@ -223,8 +223,7 @@ def push_dataset_to_hub(
test_hf_dataset = test_hf_dataset.with_format(None) test_hf_dataset = test_hf_dataset.with_format(None)
test_hf_dataset.save_to_disk(str(tests_out_dir / "train")) test_hf_dataset.save_to_disk(str(tests_out_dir / "train"))
# copy meta data to tests directory save_meta_data(info, stats, episode_data_index, tests_meta_data_dir)
shutil.copytree(meta_data_dir, tests_meta_data_dir)
# copy videos of first episode to tests directory # copy videos of first episode to tests directory
episode_index = 0 episode_index = 0
@ -233,7 +232,8 @@ def push_dataset_to_hub(
fname = f"{key}_episode_{episode_index:06d}.mp4" fname = f"{key}_episode_{episode_index:06d}.mp4"
shutil.copy(videos_dir / fname, tests_videos_dir / fname) shutil.copy(videos_dir / fname, tests_videos_dir / fname)
if not save_to_disk: if not save_to_disk and out_dir.exists():
# remove possible temporary files remaining in the output directory
shutil.rmtree(out_dir) shutil.rmtree(out_dir)