diff options
Diffstat (limited to 'src/mlia/utils/download.py')
-rw-r--r-- | src/mlia/utils/download.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/mlia/utils/download.py b/src/mlia/utils/download.py index c8d0b69..e00be28 100644 --- a/src/mlia/utils/download.py +++ b/src/mlia/utils/download.py @@ -46,9 +46,10 @@ def download( show_progress: bool = False, label: str | None = None, chunk_size: int = 8192, + timeout: int = 30, ) -> None: """Download the file.""" - with requests.get(url, stream=True, timeout=10.0) as resp: + with requests.get(url, stream=True, timeout=timeout) as resp: resp.raise_for_status() content_chunks = resp.iter_content(chunk_size=chunk_size) |