From 5a341eac2c3b42817ea3c909307952478b29c1bc Mon Sep 17 00:00:00 2001 From: martin-kokos <4807476+martin-kokos@users.noreply.github.com> Date: Fri, 24 May 2024 14:12:44 +0200 Subject: [PATCH] Fix "poster" dataset extraction (#3164) Co-authored-by: Martin Mokry Co-authored-by: maturk <30566358+maturk@users.noreply.github.com> --- nerfstudio/scripts/downloads/download_data.py | 1 + 1 file changed, 1 insertion(+) diff --git a/nerfstudio/scripts/downloads/download_data.py b/nerfstudio/scripts/downloads/download_data.py index 749b19d38..4d89c1b48 100644 --- a/nerfstudio/scripts/downloads/download_data.py +++ b/nerfstudio/scripts/downloads/download_data.py @@ -171,6 +171,7 @@ def download_capture_name(save_dir: Path, dataset_name: str, capture_name: str, with zipfile.ZipFile(download_path, "r") as zip_ref: zip_ref.extractall(tmp_path) inner_folders = os.listdir(tmp_path) + inner_folders = [folder for folder in inner_folders if folder != "__MACOSX"] assert len(inner_folders) == 1, f"There is more than one folder inside this zip file: {inner_folders}" folder = os.path.join(tmp_path, inner_folders[0]) shutil.rmtree(target_path)