Skip to content
Merged
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 3 additions & 6 deletions monai/bundle/scripts.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,6 @@
import re
import urllib
import warnings
import zipfile
from collections.abc import Mapping, Sequence
from functools import partial
from pathlib import Path
Expand All @@ -30,7 +29,7 @@
from torch.cuda import is_available

from monai._version import get_versions
from monai.apps.utils import _basename, download_url, extractall, get_logger
from monai.apps.utils import _basename, download_url, extractall, get_logger, _extract_zip
from monai.bundle.config_parser import ConfigParser
from monai.bundle.utils import DEFAULT_INFERENCE, DEFAULT_METADATA, merge_kv
from monai.bundle.workflows import BundleWorkflow, ConfigWorkflow
Expand Down Expand Up @@ -288,10 +287,8 @@ def _download_from_ngc_private(
if remove_prefix:
filename = _remove_ngc_prefix(filename, prefix=remove_prefix)
extract_path = download_path / f"{filename}"
with zipfile.ZipFile(zip_path, "r") as z:
z.extractall(extract_path)
logger.info(f"Writing into directory: {extract_path}.")

_extract_zip(zip_path, extract_path)
logger.info(f"Writing into directory: {extract_path}.")

def _get_ngc_token(api_key, retry=0):
"""Try to connect to NGC."""
Expand Down
Loading