Skip to content

Commit

Permalink
Optional param
Browse files Browse the repository at this point in the history
  • Loading branch information
martinRenou committed Nov 24, 2023
1 parent 979c384 commit b282a94
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/sagemaker/sklearn/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -307,7 +307,7 @@ def prepare_container_def(
return sagemaker.container_def(deploy_image, model_data_uri, deploy_env)

@validate_call_inputs
def serving_image_uri(self, region_name: str, instance_type: str, serverless_inference_config: Optional[ServerlessInferenceConfig] = None):
def serving_image_uri(self, region_name: str, instance_type: Optional[str] = None, serverless_inference_config: Optional[ServerlessInferenceConfig] = None):
"""Create a URI for the serving image.
Args:
Expand Down

0 comments on commit b282a94

Please sign in to comment.