Skip to content

Fix/wait respects sagemaker session 5765#5774

Open
aviruthen wants to merge 3 commits intoaws:masterfrom
aviruthen:fix/wait-respects-sagemaker-session-5765
Open

Fix/wait respects sagemaker session 5765#5774
aviruthen wants to merge 3 commits intoaws:masterfrom
aviruthen:fix/wait-respects-sagemaker-session-5765

Conversation

@aviruthen
Copy link
Copy Markdown
Collaborator

Issue #, if available:

Bug fix for the first issue mentioned in GitHub issue 5765! Ports waiting logic from PySDK V2 to V3 and allows users to optionally wait for Processing Jobs or Training Jobs without running into the aforementioned NoCredentialsError

Description of changes:

Did local integration testing to verify job waiting was working as expected, added unit tests

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.


if wait:
training_job.wait(logs=logs)
trainer_wait(
Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Could you help me understand the reasoning for moving away from the autogenerated waiter here? I noticed we introduced a dedicated waiter for ModelCustomization tailored to those specific flows — I'm curious whether the intent is to also adopt custom waiters across all high-level abstractions going forward?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants