File tree Expand file tree Collapse file tree 2 files changed +2
-1
lines changed
src/sagemaker/serve/builder
tests/unit/sagemaker/serve/builder Expand file tree Collapse file tree 2 files changed +2
-1
lines changed Original file line number Diff line number Diff line change @@ -1621,7 +1621,6 @@ def deploy(
16211621 """
16221622 if not hasattr (self , "built_model" ):
16231623 raise ValueError ("Model Needs to be built before deploying" )
1624- endpoint_name = unique_name_from_base (endpoint_name )
16251624 if not inference_config : # Real-time Deployment
16261625 return self .built_model .deploy (
16271626 instance_type = self .instance_type ,
@@ -1655,6 +1654,7 @@ def deploy(
16551654 # Multi Model and MultiContainer endpoints with Inference Component
16561655 return self .built_model .deploy (
16571656 instance_type = self .instance_type ,
1657+ endpoint_name = endpoint_name ,
16581658 mode = Mode .SAGEMAKER_ENDPOINT ,
16591659 endpoint_type = EndpointType .INFERENCE_COMPONENT_BASED ,
16601660 resources = inference_config ,
Original file line number Diff line number Diff line change @@ -4080,6 +4080,7 @@ def test_neuron_configurations_rule_set(self):
40804080 "instance_type" : "ml.g5.2xlarge" ,
40814081 "mode" : Mode .SAGEMAKER_ENDPOINT ,
40824082 "endpoint_type" : EndpointType .INFERENCE_COMPONENT_BASED ,
4083+ "endpoint_name" : "test" ,
40834084 },
40844085 },
40854086 {
You can’t perform that action at this time.
0 commit comments