Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: sagemaker model package arn (#6941)

Browse Source
This commit is contained in:
HALLOUARD 2023-10-23 15:13:12 +02:00 committed by GitHub
parent b85e99463f
commit da227dbfcd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
moto/sagemaker/models.py
View File

@ -1015,9 +1015,9 @@ class ModelPackage(BaseObject):
region_name=region_name,
account_id=account_id,
_type="model-package",
_id=f"{model_package_name}/{model_package_version}"
_id=f"{model_package_name.lower()}/{model_package_version}"
if model_package_version
else model_package_name,
else model_package_name.lower(),
)
datetime_now = datetime.now(tzutc())
self.model_package_name = model_package_name

12
tests/test_sagemaker/test_sagemaker_model_packages.py
View File

@ -411,12 +411,12 @@ def test_validate_supported_realtime_inference_instance_types_should_raise_error
def test_create_model_package():
client = boto3.client("sagemaker", region_name="eu-west-1")
resp = client.create_model_package(
ModelPackageName="test-model-package",
ModelPackageName="TestModelPackage",
ModelPackageDescription="test-model-package-description",
)
assert (
resp["ModelPackageArn"]
== "arn:aws:sagemaker:eu-west-1:123456789012:model-package/test-model-package"
== "arn:aws:sagemaker:eu-west-1:123456789012:model-package/testmodelpackage"
)
@ -446,20 +446,20 @@ def test_create_model_package_in_model_package_group():
client = boto3.client("sagemaker", region_name="eu-west-1")
client.create_model_package_group(ModelPackageGroupName="test-model-package-group")
resp_version_1 = client.create_model_package(
ModelPackageName="test-model-package",
ModelPackageName="TestModelPackage",
ModelPackageGroupName="test-model-package-group",
ModelPackageDescription="test-model-package-description",
)
resp_version_2 = client.create_model_package(
ModelPackageName="test-model-package",
ModelPackageName="TestModelPackage",
ModelPackageGroupName="test-model-package-group",
ModelPackageDescription="test-model-package-description",
)
assert (
resp_version_1["ModelPackageArn"]
== "arn:aws:sagemaker:eu-west-1:123456789012:model-package/test-model-package/1"
== "arn:aws:sagemaker:eu-west-1:123456789012:model-package/testmodelpackage/1"
)
assert (
resp_version_2["ModelPackageArn"]
== "arn:aws:sagemaker:eu-west-1:123456789012:model-package/test-model-package/2"
== "arn:aws:sagemaker:eu-west-1:123456789012:model-package/testmodelpackage/2"
)