feat: increase Timeout for LMEval tasks

sheltoncyril · sheltoncyril · commit 38fefc9c00f8 · 2025-09-04T19:59:54.000+01:00
diff --git a/tests/model_explainability/lm_eval/utils.py b/tests/model_explainability/lm_eval/utils.py
@@ -102,7 +102,7 @@ def validate_lmeval_job_pod_and_logs(lmevaljob_pod: Pod) -> None:
     )
     lmevaljob_pod.wait_for_status(status=lmevaljob_pod.Status.RUNNING, timeout=Timeout.TIMEOUT_5MIN)
     try:
-        lmevaljob_pod.wait_for_status(status=Pod.Status.SUCCEEDED, timeout=Timeout.TIMEOUT_20MIN)
+        lmevaljob_pod.wait_for_status(status=Pod.Status.SUCCEEDED, timeout=Timeout.TIMEOUT_40MIN)
     except TimeoutExpiredError as e:
         raise UnexpectedFailureError("LMEval job pod failed from a running state.") from e
     if not bool(re.search(pod_success_log_regex, lmevaljob_pod.log())):

Original file line number	Diff line number	Diff line change
`@@ -102,7 +102,7 @@ def validate_lmeval_job_pod_and_logs(lmevaljob_pod: Pod) -> None:`
`102`	`102`	`)`
`103`	`103`	`lmevaljob_pod.wait_for_status(status=lmevaljob_pod.Status.RUNNING, timeout=Timeout.TIMEOUT_5MIN)`
`104`	`104`	`try:`
`105`		`- lmevaljob_pod.wait_for_status(status=Pod.Status.SUCCEEDED, timeout=Timeout.TIMEOUT_20MIN)`
	`105`	`+ lmevaljob_pod.wait_for_status(status=Pod.Status.SUCCEEDED, timeout=Timeout.TIMEOUT_40MIN)`
`106`	`106`	`except TimeoutExpiredError as e:`
`107`	`107`	`raise UnexpectedFailureError("LMEval job pod failed from a running state.") from e`
`108`	`108`	`if not bool(re.search(pod_success_log_regex, lmevaljob_pod.log())):`