remove extraced archive if flag was set (#5055)

65438e9e · Philip Meier · GitHub · 5dc1e20b · 65438e9e
Unverified Commit 65438e9e authored Dec 08, 2021 by Philip Meier Committed by GitHub Dec 08, 2021
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

torchvision/datasets/utils.py torchvision/datasets/utils.py +2 -0

No files found.
--- a/torchvision/datasets/utils.py
+++ b/torchvision/datasets/utils.py
@@ -407,6 +407,8 @@ def extract_archive(from_path: str, to_path: Optional[str] = None, remove_finish
    extractor = _ARCHIVE_EXTRACTORS[archive_type]

    extractor(from_path, to_path, compression)
+    if remove_finished:
+        os.remove(from_path)

    return to_path