Skip to content

Commit 160d5c8

Browse files
author
Raymond Liu
committed
update naming for model to follow deploy()
1 parent 3ed3d7a commit 160d5c8

File tree

4 files changed

+17
-17
lines changed

4 files changed

+17
-17
lines changed

src/sagemaker/inference_recommender/inference_recommender_mixin.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@
1515

1616
import logging
1717
import re
18-
import uuid
1918

2019
from typing import List, Dict, Optional
2120
import sagemaker
@@ -157,14 +156,20 @@ def right_size(
157156
if isinstance(self, sagemaker.model.Model) and not isinstance(
158157
self, sagemaker.model.ModelPackage
159158
):
159+
primary_container_def = self.prepare_container_def()
160160
if not self.name:
161-
unique_tail = uuid.uuid4()
162-
self.name = "SageMaker-Model-RightSized-" + str(unique_tail)
161+
self._ensure_base_name_if_needed(
162+
image_uri=primary_container_def["Image"],
163+
script_uri=self.source_dir,
164+
model_uri=self.model_data,
165+
)
166+
self._set_model_name_if_needed()
167+
163168
create_model_args = dict(
164169
name=self.name,
165170
role=self.role,
166171
container_defs=None,
167-
primary_container=self.prepare_container_def(),
172+
primary_container=primary_container_def,
168173
vpc_config=self.vpc_config,
169174
enable_network_isolation=self.enable_network_isolation(),
170175
)

src/sagemaker/session.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -4974,14 +4974,10 @@ def create_inference_recommendations_job(
49744974
"""
49754975

49764976
if model_name is None and model_package_version_arn is None:
4977-
raise ValueError(
4978-
"Please provide either model_name or model_package_version_arn, not both."
4979-
)
4977+
raise ValueError("Please provide either model_name or model_package_version_arn.")
49804978

49814979
if model_name is not None and model_package_version_arn is not None:
4982-
raise ValueError(
4983-
"Please provide either model_name or model_package_version_arn, not both."
4984-
)
4980+
raise ValueError("Please provide either model_name or model_package_version_arn.")
49854981

49864982
if not job_name:
49874983
unique_tail = uuid.uuid4()

tests/unit/sagemaker/inference_recommender/test_inference_recommender_mixin.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,6 @@
2626
IR_SUPPORTED_CONTENT_TYPES = ["text/csv"]
2727
IR_JOB_NAME = "SMPYTHONSDK-1234567891"
2828
IR_SAMPLE_INSTANCE_TYPE = "ml.c5.xlarge"
29-
IR_MODEL_NAME = "SageMaker-Model-RightSized-sample-unique-uuid"
3029

3130
IR_SAMPLE_LIST_OF_INSTANCES_HYPERPARAMETER_RANGES = [
3231
{
@@ -187,7 +186,7 @@ def test_right_size_default_with_model_name_successful(sagemaker_session, model)
187186
)
188187

189188
assert sagemaker_session.create_model.called_with(
190-
name=IR_MODEL_NAME,
189+
name=ANY,
191190
role=IR_ROLE_ARN,
192191
container_defs=None,
193192
primary_container={},
@@ -201,7 +200,7 @@ def test_right_size_default_with_model_name_successful(sagemaker_session, model)
201200
job_name=IR_JOB_NAME,
202201
job_type="Default",
203202
job_duration_in_seconds=None,
204-
model_name=IR_MODEL_NAME,
203+
model_name=ANY,
205204
model_package_version_arn=None,
206205
framework=IR_SAMPLE_FRAMEWORK,
207206
framework_version=None,
@@ -252,7 +251,7 @@ def test_right_size_advanced_list_instances_model_name_successful(sagemaker_sess
252251
job_name=IR_JOB_NAME,
253252
job_type="Advanced",
254253
job_duration_in_seconds=7200,
255-
model_name=IR_MODEL_NAME,
254+
model_name=ANY,
256255
model_package_version_arn=None,
257256
framework=IR_SAMPLE_FRAMEWORK,
258257
framework_version=None,
@@ -303,7 +302,7 @@ def test_right_size_advanced_single_instances_model_name_successful(sagemaker_se
303302
job_name=IR_JOB_NAME,
304303
job_type="Advanced",
305304
job_duration_in_seconds=7200,
306-
model_name=IR_MODEL_NAME,
305+
model_name=ANY,
307306
model_package_version_arn=None,
308307
framework=IR_SAMPLE_FRAMEWORK,
309308
framework_version=None,

tests/unit/test_session.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3396,7 +3396,7 @@ def test_create_inference_recommendations_job_advanced_model_name_happy(sagemake
33963396
def test_create_inference_recommendations_job_missing_model_name_and_pkg(sagemaker_session):
33973397
with pytest.raises(
33983398
ValueError,
3399-
match="Please provide either model_name or model_package_version_arn, not both.",
3399+
match="Please provide either model_name or model_package_version_arn.",
34003400
):
34013401
sagemaker_session.create_inference_recommendations_job(
34023402
role=IR_ROLE_ARN,
@@ -3415,7 +3415,7 @@ def test_create_inference_recommendations_job_missing_model_name_and_pkg(sagemak
34153415
def test_create_inference_recommendations_job_provided_model_name_and_pkg(sagemaker_session):
34163416
with pytest.raises(
34173417
ValueError,
3418-
match="Please provide either model_name or model_package_version_arn, not both.",
3418+
match="Please provide either model_name or model_package_version_arn.",
34193419
):
34203420
sagemaker_session.create_inference_recommendations_job(
34213421
role=IR_ROLE_ARN,

0 commit comments

Comments
 (0)