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.