fix: avoid AccessDenied error for a while on SageMaker Studio wtih do… (#4600)

* fix: avoid AccessDenied error for a while on SageMaker Studio wtih domain isolation configured.

* Chore: fix C0301 - Line too long

* Chore: fix black-check fail

---------

Co-authored-by: liujiaor <[email protected]>
Co-authored-by: EC2 Default User <[email protected]>
Co-authored-by: Erick Benitez-Ramos <[email protected]>

Author: 4 people

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(