Increase download timeout for json files (#1556)

* Increase json file download timeout from 10s to 180s
lightly-ai · Jun 11, 2024 · 4b58536 · 4b58536
1 parent 2c8076a
commit 4b58536
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lightly/api/download.py b/lightly/api/download.py
@@ -12,6 +12,7 @@
 import tqdm
 
 from lightly.api import utils
+from lightly.api.swagger_api_client import DEFAULT_API_TIMEOUT
 
 try:
     import av
@@ -533,7 +534,7 @@ def download_json_file(
     """
     request_kwargs = request_kwargs or {}
     request_kwargs.setdefault("stream", True)
-    request_kwargs.setdefault("timeout", 10)
+    request_kwargs.setdefault("timeout", DEFAULT_API_TIMEOUT)
     req = requests if session is None else session
 
     response = req.get(url, **request_kwargs)