fix: unit tests

Author: evakravi

tests/unit/sagemaker/model/test_deploy.py

@@ -159,7 +159,12 @@ def test_deploy_accelerator_type(
         accelerator_type=ACCELERATOR_TYPE,
     )
 
-    create_sagemaker_model.assert_called_with(INSTANCE_TYPE, ACCELERATOR_TYPE, None, None)
+    create_sagemaker_model.assert_called_with(
+        instance_type=INSTANCE_TYPE,
+        accelerator_type=ACCELERATOR_TYPE,
+        tags=None,
+        serverless_inference_config=None,
+    )
     production_variant.assert_called_with(
         MODEL_NAME,
         INSTANCE_TYPE,
@@ -271,7 +276,12 @@ def test_deploy_tags(create_sagemaker_model, production_variant, name_from_base,
     tags = [{"Key": "ModelName", "Value": "TestModel"}]
     model.deploy(instance_type=INSTANCE_TYPE, initial_instance_count=INSTANCE_COUNT, tags=tags)
 
-    create_sagemaker_model.assert_called_with(INSTANCE_TYPE, None, tags, None)
+    create_sagemaker_model.assert_called_with(
+        instance_type=INSTANCE_TYPE,
+        accelerator_type=None,
+        tags=tags,
+        serverless_inference_config=None,
+    )
     sagemaker_session.endpoint_from_production_variants.assert_called_with(
         name=ENDPOINT_NAME,
         production_variants=[BASE_PRODUCTION_VARIANT],
@@ -463,7 +473,12 @@ def test_deploy_serverless_inference(production_variant, create_sagemaker_model,
         serverless_inference_config=serverless_inference_config,
     )
 
-    create_sagemaker_model.assert_called_with(None, None, None, serverless_inference_config)
+    create_sagemaker_model.assert_called_with(
+        instance_type=None,
+        accelerator_type=None,
+        tags=None,
+        serverless_inference_config=serverless_inference_config,
+    )
     production_variant.assert_called_with(
         MODEL_NAME,
         None,