fix: fix format and unit tests

Author: xiaoyi-cheng

tests/integ/test_explainer.py

@@ -81,9 +81,7 @@ def endpoint_name(sagemaker_session):
 
 
 def test_describe_explainer_config(sagemaker_session, endpoint_name):
-    endpoint_desc = sagemaker_session.sagemaker_client.describe_endpoint(
-        EndpointName=endpoint_name
-    )
+    endpoint_desc = sagemaker_session.sagemaker_client.describe_endpoint(EndpointName=endpoint_name)
 
     endpoint_config_desc = sagemaker_session.sagemaker_client.describe_endpoint_config(
         EndpointConfigName=endpoint_desc["EndpointConfigName"]

tests/unit/sagemaker/inference_recommender/test_inference_recommender_mixin.py

@@ -565,6 +565,7 @@ def test_deploy_right_size_with_model_package_succeeds(
     sagemaker_session.endpoint_from_production_variants.assert_called_with(
         async_inference_config_dict=None,
         data_capture_config_dict=None,
+        explainer_config_dict=None,
         kms_key=None,
         name="ir-endpoint-test",
         production_variants=IR_PRODUCTION_VARIANTS,
@@ -586,6 +587,7 @@ def test_deploy_right_size_with_both_overrides_succeeds(
     sagemaker_session.endpoint_from_production_variants.assert_called_with(
         async_inference_config_dict=None,
         data_capture_config_dict=None,
+        explainer_config_dict=None,
         kms_key=None,
         name="ir-endpoint-test",
         production_variants=IR_OVERRIDDEN_PRODUCTION_VARIANTS,
@@ -638,6 +640,7 @@ def test_deploy_right_size_serverless_override(sagemaker_session, default_right_
         wait=True,
         data_capture_config_dict=None,
         async_inference_config_dict=None,
+        explainer_config_dict=None,
     )
 
 
@@ -659,6 +662,7 @@ def test_deploy_right_size_async_override(sagemaker_session, default_right_sized
         wait=True,
         data_capture_config_dict=None,
         async_inference_config_dict={"OutputConfig": {"S3OutputPath": "s3://some-path"}},
+        explainer_config_dict=None,
     )
 
 

tests/unit/test_estimator.py

@@ -2789,6 +2789,7 @@ def test_fit_deploy_tags_in_estimator(name_from_base, sagemaker_session):
         wait=True,
         data_capture_config_dict=None,
         async_inference_config_dict=None,
+        explainer_config_dict=None,
     )
 
     sagemaker_session.create_model.assert_called_with(
@@ -2839,6 +2840,7 @@ def test_fit_deploy_tags(name_from_base, sagemaker_session):
         wait=True,
         data_capture_config_dict=None,
         async_inference_config_dict=None,
+        explainer_config_dict=None,
     )
 
     sagemaker_session.create_model.assert_called_with(