diff --git a/src/sagemaker/session.py b/src/sagemaker/session.py index 4209f37ae6..fb65c9f2cf 100644 --- a/src/sagemaker/session.py +++ b/src/sagemaker/session.py @@ -8348,7 +8348,9 @@ def _logs_for_job( # noqa: C901 - suppress complexity warning for this method """ sagemaker_client = sagemaker_session.sagemaker_client request_end_time = time.time() + timeout if timeout else None - description = sagemaker_client.describe_training_job(TrainingJobName=job_name) + description = _wait_until( + lambda: sagemaker_client.describe_training_job(TrainingJobName=job_name) + ) print(secondary_training_status_message(description, None), end="") instance_count, stream_names, positions, client, log_group, dot, color_wrap = _logs_init(