update attribute param to be correct

shreyapandit · shreyapandit · commit 1737f9ac5733 · 2021-08-27T10:34:33.000-07:00
diff --git a/src/sagemaker/automl/automl.py b/src/sagemaker/automl/automl.py
@@ -420,7 +420,7 @@ def deploy(
             serializer=serializer,
             deserializer=deserializer,
             endpoint_name=endpoint_name,
-            model_kms_key=model_kms_key,
+            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
@@ -595,7 +595,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,
+        kms_key=OUTPUT_KMS_KEY,
         tags=TAGS,
         wait=False,
     )

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