From 55aa0bf800bae132648f89a1a64d00bdb50f9ad6 Mon Sep 17 00:00:00 2001
From: Joshua Towner <josh@towner.xyz>
Date: Mon, 4 May 2026 07:42:13 -0700
Subject: [PATCH] fix: increase default timeout for training jobs

---
 .../src/sagemaker/train/common_utils/trainer_wait.py            | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/sagemaker-train/src/sagemaker/train/common_utils/trainer_wait.py b/sagemaker-train/src/sagemaker/train/common_utils/trainer_wait.py
index a7b41b5b1e..a4bf886472 100644
--- a/sagemaker-train/src/sagemaker/train/common_utils/trainer_wait.py
+++ b/sagemaker-train/src/sagemaker/train/common_utils/trainer_wait.py
@@ -200,7 +200,7 @@ def get_mlflow_url(training_job) -> str:
 def wait(
         training_job: TrainingJob,
         poll: int = 5,
-        timeout: Optional[int] = 3000
+        timeout: Optional[int] = 43200
 ) -> None:
     """Wait for training job to complete with progress tracking.