Skip to content

Commit f64ca6d

Browse files
committed
fix: tags for ModelPackage
1 parent c1157bd commit f64ca6d

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

src/sagemaker/model.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1375,7 +1375,10 @@ def deploy(
13751375
self._base_name = "-".join((self._base_name, compiled_model_suffix))
13761376

13771377
self._create_sagemaker_model(
1378-
instance_type, accelerator_type, tags, serverless_inference_config
1378+
instance_type=instance_type,
1379+
accelerator_type=accelerator_type,
1380+
tags=tags,
1381+
serverless_inference_config=serverless_inference_config,
13791382
)
13801383

13811384
serverless_inference_config_dict = (

tests/unit/sagemaker/model/test_model_package.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -197,7 +197,7 @@ def test_create_sagemaker_model_include_tags(sagemaker_session):
197197
sagemaker_session=sagemaker_session,
198198
)
199199

200-
model_package._create_sagemaker_model(tags=tags)
200+
model_package.deploy(tags=tags, instance_type="ml.p2.xlarge", initial_instance_count=1)
201201

202202
sagemaker_session.create_model.assert_called_with(
203203
model_name,

0 commit comments

Comments
 (0)