From 4f5cc28b8d82f69434403aae74a60dfe0677a736 Mon Sep 17 00:00:00 2001 From: David McAllister Date: Tue, 4 Oct 2022 12:59:06 -0700 Subject: [PATCH] Data download script fix (#646) * Fixed case where no data has been downloaded * Directory deletion --- scripts/downloads/download_data.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/scripts/downloads/download_data.py b/scripts/downloads/download_data.py index 2577e8683..6ee4eab32 100755 --- a/scripts/downloads/download_data.py +++ b/scripts/downloads/download_data.py @@ -21,7 +21,8 @@ def download_blender(save_dir: Path): blender_file_id = "18JxhpWD-4ZmuFKLzKlAw-w5PpzZxXOcG" final_path = save_dir / Path("blender") - os.remove(str(final_path)) + if os.path.exists(final_path): + shutil.rmtree(str(final_path)) url = f"https://drive.google.com/uc?id={blender_file_id}" download_path = save_dir / "blender_data.zip" gdown.download(url, output=str(download_path))