-
Notifications
You must be signed in to change notification settings - Fork 1.2k
fix: removing kwargs as this is breaking predictor_cls param for mode… #4816
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
src/sagemaker/model.py
Outdated
@@ -1696,8 +1696,7 @@ def deploy( | |||
instance_type=instance_type, | |||
accelerator_type=accelerator_type, | |||
tags=tags, | |||
serverless_inference_config=serverless_inference_config, | |||
**kwargs, | |||
serverless_inference_config=serverless_inference_config |
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.
oof, this looks like a backward incompatible change, are you sure we can simply do that?
We have to explicitly list the arguments that are eligible to be passed here.
@@ -1697,7 +1700,8 @@ def deploy( | |||
accelerator_type=accelerator_type, | |||
tags=tags, | |||
serverless_inference_config=serverless_inference_config, | |||
**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.
optional: maybe log any kwargs that are being ignored?
@@ -771,7 +771,7 @@ def test_jumpstart_model_kwargs_match_parent_class(self): | |||
and reach out to JumpStart team.""" | |||
|
|||
init_args_to_skip: Set[str] = set(["model_reference_arn"]) | |||
deploy_args_to_skip: Set[str] = set(["kwargs"]) | |||
deploy_args_to_skip: Set[str] = set(["kwargs", "model_reference_arn"]) |
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.
why don't we accept this argument for JumpStartModel.deploy
?
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.
+1
…l deploy
Issue #, if available:
Description of changes:
predictor_cls
is not expected as an argument to_create_sagemaker_model
.Testing done:
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.General
Tests
unique_name_from_base
to create resource names in integ tests (if appropriate)By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.