Update more methods on AutoML test

shreyapandit · shreyapandit · commit 99850a5eb216 · 2021-08-27T10:02:36.000-07:00
diff --git a/tests/unit/sagemaker/automl/test_auto_ml.py b/tests/unit/sagemaker/automl/test_auto_ml.py
@@ -541,6 +541,7 @@ def test_deploy(sagemaker_session, candidate_mock):
         initial_instance_count=INSTANCE_COUNT,
         instance_type=INSTANCE_TYPE,
         sagemaker_session=sagemaker_session,
+        model_kms_key=OUTPUT_KMS_KEY,
     )
     auto_ml.create_model.assert_called_once()
     mock_pipeline.deploy.assert_called_once()

Original file line number	Diff line number	Diff line change
`@@ -541,6 +541,7 @@ def test_deploy(sagemaker_session, candidate_mock):`
`541`	`541`	`initial_instance_count=INSTANCE_COUNT,`
`542`	`542`	`instance_type=INSTANCE_TYPE,`
`543`	`543`	`sagemaker_session=sagemaker_session,`
	`544`	`+ model_kms_key=OUTPUT_KMS_KEY,`
`544`	`545`	`)`
`545`	`546`	`auto_ml.create_model.assert_called_once()`
`546`	`547`	`mock_pipeline.deploy.assert_called_once()`