From 94490d754c6b45ff5a748cccb6e4331236f4c467 Mon Sep 17 00:00:00 2001 From: Mike VanDenburgh Date: Wed, 26 Jun 2024 13:44:03 -0400 Subject: [PATCH] Add timeout to gitlab requests in job classification task --- analytics/analytics/core/job_failure_classifier/__init__.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/analytics/analytics/core/job_failure_classifier/__init__.py b/analytics/analytics/core/job_failure_classifier/__init__.py index 4f4425b1c..62dc1981d 100644 --- 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