-
Notifications
You must be signed in to change notification settings - Fork 1.2k
Improving endpoint deletion #9
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
tests/integ/test_cifar.py
Outdated
@@ -0,0 +1,45 @@ | |||
# Copyright 2017 Amazon.com, Inc. or its affiliates. All Rights Reserved. |
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.
rename file to test_tf_cifar.py
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.
lgtm. Just one small suggestion (and also Jonathan's comment about the filename)
tests/integ/timeout.py
Outdated
yield [t] | ||
finally: | ||
sagemaker_session.delete_endpoint(endpoint_name) | ||
LOGGER.info('deleted endpoint') |
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.
Adding the endpoint name to the logging statement could be helpful
Add job base class
FrameworkProcessor: revert `run(entry_point=...)` back to `run(code=...)`
timeout_and_delete_endpoint_by_name
context manager to easily delete an endpoint in the end of the integ test