File tree 2 files changed +15
-1
lines changed
2 files changed +15
-1
lines changed Original file line number Diff line number Diff line change @@ -3222,7 +3222,9 @@ def create_model_package_from_containers(
3222
3222
)
3223
3223
3224
3224
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
+ ):
3226
3228
_create_resource (
3227
3229
lambda : self .sagemaker_client .create_model_package_group (
3228
3230
ModelPackageGroupName = request ["ModelPackageGroupName" ]
Original file line number Diff line number Diff line change @@ -2650,6 +2650,18 @@ def test_create_model_package_from_containers(sagemaker_session):
2650
2650
sagemaker_session .sagemaker_client .create_model_package .assert_called_once ()
2651
2651
2652
2652
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
+
2653
2665
def test_create_model_package_from_containers_name_conflict (sagemaker_session ):
2654
2666
model_package_name = "sagemaker-model-package"
2655
2667
model_package_group_name = "sagemaker-model-package-group"
You can’t perform that action at this time.
0 commit comments