Skip to content

Commit 5482c6d

Browse files
fix: avoid AccessDenied error for a while on SageMaker Studio wtih domain isolation configured.
1 parent 48234ae commit 5482c6d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/sagemaker/session.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7884,7 +7884,7 @@ def _logs_for_job( # noqa: C901 - suppress complexity warning for this method
78847884
"""
78857885
sagemaker_client = sagemaker_session.sagemaker_client
78867886
request_end_time = time.time() + timeout if timeout else None
7887-
description = sagemaker_client.describe_training_job(TrainingJobName=job_name)
7887+
description = _wait_until(lambda: sagemaker_client.describe_training_job(TrainingJobName=job_name))
78887888
print(secondary_training_status_message(description, None), end="")
78897889

78907890
instance_count, stream_names, positions, client, log_group, dot, color_wrap = _logs_init(

0 commit comments

Comments
 (0)