Skip to content

Commit d6bb418

Browse files
committed
add ut & indentation for cross_account mpg
1 parent c81c103 commit d6bb418

File tree

2 files changed

+15
-1
lines changed

2 files changed

+15
-1
lines changed

src/sagemaker/session.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -3222,7 +3222,9 @@ def create_model_package_from_containers(
32223222
)
32233223

32243224
def submit(request):
3225-
if model_package_group_name is not None and not model_package_group_name.startswith("arn:"):
3225+
if model_package_group_name is not None and not model_package_group_name.startswith(
3226+
"arn:"
3227+
):
32263228
_create_resource(
32273229
lambda: self.sagemaker_client.create_model_package_group(
32283230
ModelPackageGroupName=request["ModelPackageGroupName"]

tests/unit/test_session.py

+12
Original file line numberDiff line numberDiff line change
@@ -2650,6 +2650,18 @@ def test_create_model_package_from_containers(sagemaker_session):
26502650
sagemaker_session.sagemaker_client.create_model_package.assert_called_once()
26512651

26522652

2653+
def test_create_model_package_from_containers_cross_account_mpg_name(sagemaker_session):
2654+
mpg_name = "arn:aws:sagemaker:us-east-1:215995503607:model-package-group/stage-dev"
2655+
content_types = ["text/csv"]
2656+
response_types = ["text/csv"]
2657+
sagemaker_session.create_model_package_from_containers(
2658+
model_package_group_name=mpg_name,
2659+
content_types=content_types,
2660+
response_types=response_types,
2661+
)
2662+
sagemaker_session.sagemaker_client.create_model_package.assert_called_once()
2663+
2664+
26532665
def test_create_model_package_from_containers_name_conflict(sagemaker_session):
26542666
model_package_name = "sagemaker-model-package"
26552667
model_package_group_name = "sagemaker-model-package-group"

0 commit comments

Comments
 (0)