fix: add metadata_properties args (aws#537)

Author: chuyang-deng

src/sagemaker/session.py

@@ -2557,6 +2557,7 @@ def create_model_package_from_containers(
         model_package_name=None,
         model_package_group_name=None,
         model_metrics=None,
+        metadata_properties=None,
         marketplace_cert=False,
         approval_status="PendingManualApproval",
         description=None,
@@ -2578,6 +2579,7 @@ def create_model_package_from_containers(
                 `model_package_name`, using `model_package_group_name` makes the Model Package
                 versioned (default: None).
             model_metrics (ModelMetrics): ModelMetrics object (default: None).
+            metadata_properties (MetadataProperties): MetadataProperties object (default: None)
             marketplace_cert (bool): A boolean value indicating if the Model Package is certified
                 for AWS Marketplace (default: False).
             approval_status (str): Model Approval Status, values can be "Approved", "Rejected",
@@ -2594,6 +2596,7 @@ def create_model_package_from_containers(
             inference_instances,
             transform_instances,
             model_metrics,
+            metadata_properties,
             marketplace_cert,
             approval_status,
             description,
@@ -2610,6 +2613,7 @@ def _get_create_model_package_request(
         inference_instances=None,
         transform_instances=None,
         model_metrics=None,
+        metadata_properties=None,
         marketplace_cert=False,
         approval_status="PendingManualApproval",
         description=None,
@@ -2631,6 +2635,7 @@ def _get_create_model_package_request(
             transform_instances (list): A list of the instance types on which a transformation
                 job can be run or on which an endpoint can be deployed (default: None).
             model_metrics (ModelMetrics): ModelMetrics object (default: None).
+            metadata_properties (MetadataProperties): MetadataProperties object (default: None).
             marketplace_cert (bool): A boolean value indicating if the Model Package is certified
                 for AWS Marketplace (default: False).
             approval_status (str): Model Approval Status, values can be "Approved", "Rejected",
@@ -2651,6 +2656,8 @@ def _get_create_model_package_request(
             request_dict["ModelPackageDescription"] = description
         if model_metrics:
             request_dict["ModelMetrics"] = model_metrics
+        if metadata_properties:
+            request_dict["MetadataProperties"] = metadata_properties
         if containers is not None:
             if not all([content_types, response_types, inference_instances, transform_instances]):
                 raise ValueError(

tests/unit/test_session.py

@@ -2205,6 +2205,12 @@ def test_create_model_package_from_containers_all_args(sagemaker_session):
             "S3Uri": "s3://...",
         }
     }
+    metadata_properties = {
+        "CommitId": "test-commit-id",
+        "Repository": "test-repository",
+        "GeneratedBy": "sagemaker-python-sdk",
+        "ProjectId": "unit-test",
+    }
     marketplace_cert = (True,)
     approval_status = ("Approved",)
     description = "description"
@@ -2216,6 +2222,7 @@ def test_create_model_package_from_containers_all_args(sagemaker_session):
         transform_instances=transform_instances,
         model_package_name=model_package_name,
         model_metrics=model_metrics,
+        metadata_properties=metadata_properties,
         marketplace_cert=marketplace_cert,
         approval_status=approval_status,
         description=description,
@@ -2231,6 +2238,7 @@ def test_create_model_package_from_containers_all_args(sagemaker_session):
         },
         "ModelPackageDescription": description,
         "ModelMetrics": model_metrics,
+        "MetadataProperties": metadata_properties,
         "CertifyForMarketplace": marketplace_cert,
         "ModelApprovalStatus": approval_status,
     }