From 07688ec1f7e72f53956a8f7d9d0fc3c602afe98d Mon Sep 17 00:00:00 2001 From: sailesh duddupudi Date: Thu, 9 Jan 2025 20:54:36 +0000 Subject: [PATCH] Increase timeouts Signed-off-by: sailesh duddupudi --- sdk/python/test/e2e/test_e2e_jaxjob.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sdk/python/test/e2e/test_e2e_jaxjob.py b/sdk/python/test/e2e/test_e2e_jaxjob.py index eba4fa9842..c60a87554d 100644 --- a/sdk/python/test/e2e/test_e2e_jaxjob.py +++ b/sdk/python/test/e2e/test_e2e_jaxjob.py @@ -88,7 +88,7 @@ def test_sdk_e2e_with_gang_scheduling(job_namespace): logging.info(TRAINING_CLIENT.list_jobs(job_namespace)) try: - utils.verify_job_e2e(TRAINING_CLIENT, JOB_NAME, job_namespace, wait_timeout=6000) + utils.verify_job_e2e(TRAINING_CLIENT, JOB_NAME, job_namespace, wait_timeout=3000) except Exception as e: utils.print_job_results(TRAINING_CLIENT, JOB_NAME, job_namespace) TRAINING_CLIENT.delete_job(JOB_NAME, job_namespace) @@ -123,7 +123,7 @@ def test_sdk_e2e(job_namespace): logging.info(TRAINING_CLIENT.list_jobs(job_namespace)) try: - utils.verify_job_e2e(TRAINING_CLIENT, JOB_NAME, job_namespace, wait_timeout=900) + utils.verify_job_e2e(TRAINING_CLIENT, JOB_NAME, job_namespace, wait_timeout=3000) except Exception as e: utils.print_job_results(TRAINING_CLIENT, JOB_NAME, job_namespace) TRAINING_CLIENT.delete_job(JOB_NAME, job_namespace)