Skip to content

Commit 7a6d1ce

Browse files
committed
Fix multiple IC support in modelBuilder
1 parent d395adc commit 7a6d1ce

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

src/sagemaker/serve/builder/model_builder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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,

tests/unit/sagemaker/serve/builder/test_model_builder.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff 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
{

0 commit comments

Comments
 (0)