From 38f99e9d37efcbfbdbaf6f4d6861b1c0fffae210 Mon Sep 17 00:00:00 2001 From: pintaoz Date: Tue, 13 May 2025 10:19:09 -0700 Subject: [PATCH] Fix test_deploy_with_update_endpoint() --- tests/unit/sagemaker/model/test_deploy.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/unit/sagemaker/model/test_deploy.py b/tests/unit/sagemaker/model/test_deploy.py index 4167ca62c3..ef8b5e6af5 100644 --- a/tests/unit/sagemaker/model/test_deploy.py +++ b/tests/unit/sagemaker/model/test_deploy.py @@ -1090,7 +1090,7 @@ def test_deploy_with_update_endpoint(production_variant, name_from_base, sagemak container_startup_health_check_timeout=None, explainer_config_dict=None, async_inference_config_dict=None, - serverless_inference_config=None, + serverless_inference_config_dict=None, routing_config=None, inference_ami_version=None, ) @@ -1124,7 +1124,7 @@ def test_deploy_with_update_endpoint(production_variant, name_from_base, sagemak container_startup_health_check_timeout=None, explainer_config_dict=None, async_inference_config_dict=None, - serverless_inference_config=serverless_inference_config_dict, + serverless_inference_config_dict=serverless_inference_config_dict, routing_config=None, inference_ami_version=None, ) @@ -1164,7 +1164,7 @@ def test_deploy_with_update_endpoint(production_variant, name_from_base, sagemak container_startup_health_check_timeout=None, explainer_config_dict=None, async_inference_config_dict=async_inference_config_dict, - serverless_inference_config=None, + serverless_inference_config_dict=None, routing_config=None, inference_ami_version=None, )