-
Notifications
You must be signed in to change notification settings - Fork 1.2k
simplify create_image_uri function #462
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
Codecov Report
@@ Coverage Diff @@
## master #462 +/- ##
==========================================
+ Coverage 93.73% 93.87% +0.14%
==========================================
Files 56 56
Lines 4196 4199 +3
==========================================
+ Hits 3933 3942 +9
+ Misses 263 257 -6
Continue to review full report at Codecov.
|
|
||
def create_image_uri(region, framework, instance_type, framework_version, py_version, account='520713654638', | ||
|
||
def create_image_uri(region, framework, instance_type, framework_version, py_version=None, account='520713654638', |
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.
Thanks for doing this.
tests/unit/test_tfs.py
Outdated
for k in ckwargs: | ||
assert kwargs[k] == ckwargs[k] | ||
|
||
# sagemaker_session.sagemaker_runtime_client.invoke_endpoint.assert_called_once_with(**kwargs) |
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.
Remove the comment. Is it possible to add a docstring just to clarify a little bit of the function does?
It took me a while to understand.
Description of changes:
simplify create_image_uri function
add failure test case for tensorflow serving predictor
Merge Checklist
Put an
x
in the boxes that apply. You can also fill these out after creating the PR. If you're unsure about any of them, don't hesitate to ask. We're here to help! This is simply a reminder of what we are going to look for before merging your pull request.By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.