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(
1621
1621
"""
1622
1622
if not hasattr (self , "built_model" ):
1623
1623
raise ValueError ("Model Needs to be built before deploying" )
1624
- endpoint_name = unique_name_from_base (endpoint_name )
1625
1624
if not inference_config : # Real-time Deployment
1626
1625
return self .built_model .deploy (
1627
1626
instance_type = self .instance_type ,
@@ -1655,6 +1654,7 @@ def deploy(
1655
1654
# Multi Model and MultiContainer endpoints with Inference Component
1656
1655
return self .built_model .deploy (
1657
1656
instance_type = self .instance_type ,
1657
+ endpoint_name = endpoint_name ,
1658
1658
mode = Mode .SAGEMAKER_ENDPOINT ,
1659
1659
endpoint_type = EndpointType .INFERENCE_COMPONENT_BASED ,
1660
1660
resources = inference_config ,
Original file line number Diff line number Diff line change @@ -4080,6 +4080,7 @@ def test_neuron_configurations_rule_set(self):
4080
4080
"instance_type" : "ml.g5.2xlarge" ,
4081
4081
"mode" : Mode .SAGEMAKER_ENDPOINT ,
4082
4082
"endpoint_type" : EndpointType .INFERENCE_COMPONENT_BASED ,
4083
+ "endpoint_name" : "test" ,
4083
4084
},
4084
4085
},
4085
4086
{
You can’t perform that action at this time.
0 commit comments