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, region_name=region_name,
account_id=account_id, account_id=account_id,
_type="model-package", _type="model-package",
_id=f"{model_package_name}/{model_package_version}" _id=f"{model_package_name.lower()}/{model_package_version}"
if model_package_version if model_package_version
else model_package_name, else model_package_name.lower(),
) )
datetime_now = datetime.now(tzutc()) datetime_now = datetime.now(tzutc())
self.model_package_name = model_package_name 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(): def test_create_model_package():
client = boto3.client("sagemaker", region_name="eu-west-1") client = boto3.client("sagemaker", region_name="eu-west-1")
resp = client.create_model_package( resp = client.create_model_package(
ModelPackageName="test-model-package", ModelPackageName="TestModelPackage",
ModelPackageDescription="test-model-package-description", ModelPackageDescription="test-model-package-description",
) )
assert ( assert (
resp["ModelPackageArn"] 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 = boto3.client("sagemaker", region_name="eu-west-1")
client.create_model_package_group(ModelPackageGroupName="test-model-package-group") client.create_model_package_group(ModelPackageGroupName="test-model-package-group")
resp_version_1 = client.create_model_package( resp_version_1 = client.create_model_package(
ModelPackageName="test-model-package", ModelPackageName="TestModelPackage",
ModelPackageGroupName="test-model-package-group", ModelPackageGroupName="test-model-package-group",
ModelPackageDescription="test-model-package-description", ModelPackageDescription="test-model-package-description",
) )
resp_version_2 = client.create_model_package( resp_version_2 = client.create_model_package(
ModelPackageName="test-model-package", ModelPackageName="TestModelPackage",
ModelPackageGroupName="test-model-package-group", ModelPackageGroupName="test-model-package-group",
ModelPackageDescription="test-model-package-description", ModelPackageDescription="test-model-package-description",
) )
assert ( assert (
resp_version_1["ModelPackageArn"] 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 ( assert (
resp_version_2["ModelPackageArn"] 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"
) )