Propagate KMS key to model.deploy

shreyapandit · shreyapandit · commit 1a414ea5ca88 · 2021-08-27T09:54:01.000-07:00
diff --git a/src/sagemaker/automl/automl.py b/src/sagemaker/automl/automl.py
@@ -420,6 +420,7 @@ def deploy(
             serializer=serializer,
             deserializer=deserializer,
             endpoint_name=endpoint_name,
+            model_kms_key=model_kms_key,
             tags=tags,
             wait=wait,
         )
diff --git a/tests/unit/sagemaker/automl/test_auto_ml.py b/tests/unit/sagemaker/automl/test_auto_ml.py
@@ -594,6 +594,7 @@ def test_deploy_optional_args(candidate_estimator, sagemaker_session, candidate_
         serializer=None,
         deserializer=None,
         endpoint_name=JOB_NAME,
+        model_kms_key=OUTPUT_KMS_KEY,
         tags=TAGS,
         wait=False,
     )

Original file line number	Diff line number	Diff line change
`@@ -594,6 +594,7 @@ def test_deploy_optional_args(candidate_estimator, sagemaker_session, candidate_`
`594`	`594`	`serializer=None,`
`595`	`595`	`deserializer=None,`
`596`	`596`	`endpoint_name=JOB_NAME,`
	`597`	`+ model_kms_key=OUTPUT_KMS_KEY,`
`597`	`598`	`tags=TAGS,`
`598`	`599`	`wait=False,`
`599`	`600`	`)`