-
Notifications
You must be signed in to change notification settings - Fork 16.3k
deferrable mode for SageMakerTuningOperator and SageMakerEndpointOperator
#32112
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
| max_attempts: int | None = None, | ||
| max_attempts: int = 480, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
passing None was not working before, so this is not a breaking change since it was already broken.
| cp /root/.aws/credentials /tmp/credentials && | ||
| # login to public ecr repo containing amazonlinux image | ||
| docker login --username {creds.username} --password {creds.password} public.ecr.aws | ||
| docker login --username {creds.username} --password {creds.password} public.ecr.aws && |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
the old version was working too, but at least this homogeneous with the other lines of the command
| rm /tmp/credentials && | ||
| # login again, this time to the private repo we created to hold that specific image | ||
| aws ecr get-login-password --region {ecr_region} | |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
this is probably from before, when we didn't pass credentials in the command line, but it's now useless.
| "Failed to prepare docker image for the preprocessing job.\n" | ||
| f"The following error happened while executing the sequence of bash commands:\n{stderr}" | ||
| "The following error happened while executing the sequence of bash commands:\n" | ||
| f"{stderr.decode()}" |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
stderr is a byte array. Without the decode(), the \n are not escaped for instance, and the whole output appears on one line, which is not the best
SageMakerTuningOperatorSageMakerTuningOperator and SageMakerEndpointOperator
SageMakerTuningOperator and SageMakerEndpointOperatorSageMakerTuningOperator and SageMakerEndpointOperator
SageMakerTuningOperator and SageMakerEndpointOperator SageMakerTuningOperator and SageMakerEndpointOperator
|
TIL: when updating a PR title with |
took the opportunity to fix tiny things in the system test,
and also migrate to the trigger to the method that logs the status while it waits