Fixed test

ivanmkc · ivanmkc · commit 035197f12816 · 2021-04-26T20:08:26.000-04:00
diff --git a/samples/model-builder/deploy_model_with_automatic_resources_test.py b/samples/model-builder/deploy_model_with_automatic_resources_test.py
@@ -18,10 +18,7 @@
 
 
 def test_deploy_model_with_automatic_resources_sample(
-    mock_sdk_init,
-    mock_model,
-    mock_init_model,
-    mock_deploy_model_with_automatic_resources,
+    mock_sdk_init, mock_model, mock_init_model, mock_deploy_model,
 ):
 
     deploy_model_with_automatic_resources_sample.deploy_model_with_automatic_resources_sample(
@@ -43,7 +40,7 @@ def test_deploy_model_with_automatic_resources_sample(
 
     mock_init_model.assert_called_once_with(model_name=constants.MODEL_NAME)
 
-    mock_deploy_model_with_automatic_resources.assert_called_once_with(
+    mock_deploy_model.assert_called_once_with(
         endpoint=constants.ENDPOINT_NAME,
         deployed_model_display_name=constants.DEPLOYED_MODEL_DISPLAY_NAME,
         traffic_percentage=constants.TRAFFIC_PERCENTAGE,
