Merge branch 'master' into chore/telemetry-for-deployment-configs

Author: evakravi

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(