aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/utils/download.py
diff options
context:
space:
mode:
authorBenjamin Klimczak <benjamin.klimczak@arm.com>2022-11-07 12:57:15 +0000
committerBenjamin Klimczak <benjamin.klimczak@arm.com>2022-11-11 12:10:26 +0000
commitce9b17650d024886b24ad820f0f1815fc23b19f3 (patch)
treea7d113f751b8856aabcd021464edec16e23ba6f8 /src/mlia/utils/download.py
parente40a7adadd254e29d71af38f69a0a20ff4871eef (diff)
downloadmlia-ce9b17650d024886b24ad820f0f1815fc23b19f3.tar.gz
MLIA-701 Update dependencies
- Update TensorFlow dependencies for x86_64 - Adapt unit tests to new TensorFlow version - Update linters (including pre-commit hooks) and fix issues - Use conditional import to fix tflite compat code for aarch64 Change-Id: I1a9b080b900ab65e38f7f2552562822bbfdcd259
Diffstat (limited to 'src/mlia/utils/download.py')
-rw-r--r--src/mlia/utils/download.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/src/mlia/utils/download.py b/src/mlia/utils/download.py
index 9ef2d9e..c8d0b69 100644
--- a/src/mlia/utils/download.py
+++ b/src/mlia/utils/download.py
@@ -48,7 +48,7 @@ def download(
chunk_size: int = 8192,
) -> None:
"""Download the file."""
- with requests.get(url, stream=True) as resp:
+ with requests.get(url, stream=True, timeout=10.0) as resp:
resp.raise_for_status()
content_chunks = resp.iter_content(chunk_size=chunk_size)