Unpack .tgz files to target_path dir

Author: slice4e

benchmark/dataset.py

@@ -195,9 +195,11 @@ def _download_file(self, relative_path: str, url: str):
         self._extract_or_move_file(tmp_path, target_path)
 
     def _extract_or_move_file(self, tmp_path, target_path):
+        # If the file is a .tgz or .tar.gz archive, always extract to a directory named target_path
         if tmp_path.endswith(".tgz") or tmp_path.endswith(".tar.gz"):
-            print(f"Extracting: {tmp_path} -> {target_path}")
-            (DATASETS_DIR / self.config.path).mkdir(exist_ok=True, parents=True)
+            print(f"Extracting: {tmp_path} -> {target_path}/")
+            # Ensure target_path is a directory
+            Path(target_path).mkdir(exist_ok=True, parents=True)
             with tarfile.open(tmp_path) as file:
                 file.extractall(target_path)
             os.remove(tmp_path)