From d8f85300017f05ee01397abc89494e889b80c62c Mon Sep 17 00:00:00 2001
From: Philip Meier <github.pmeier@posteo.de>
Date: Wed, 8 Dec 2021 09:12:49 +0100
Subject: [PATCH] remove extraced archive if flag was set

---
 torchvision/datasets/utils.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/torchvision/datasets/utils.py b/torchvision/datasets/utils.py
index cdeacb37c6d..dbc9cf2a6b4 100644
--- 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