Add timeout to gitlab requests in job classification task

spack · Jun 26, 2024 · 94490d7 · 94490d7
1 parent dfe5db2
commit 94490d7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/analytics/analytics/core/job_failure_classifier/__init__.py b/analytics/analytics/core/job_failure_classifier/__init__.py
@@ -181,7 +181,10 @@ def _collect_pod_status(job_input_data: dict[str, Any], job_trace: str):
 )
 def upload_job_failure_classification(job_input_data_json: str) -> None:
     gl = gitlab.Gitlab(
-        settings.GITLAB_ENDPOINT, settings.GITLAB_TOKEN, retry_transient_errors=True
+        settings.GITLAB_ENDPOINT,
+        settings.GITLAB_TOKEN,
+        retry_transient_errors=True,
+        timeout=15,
     )
 
     # Read input data and extract params