From d889212ffa62bd54e781a4a4c990a99a42f0803b Mon Sep 17 00:00:00 2001 From: Shreya Pandit Date: Fri, 20 May 2022 14:31:02 -0700 Subject: [PATCH 1/2] Use Async Inference Config when available for endpoint update --- src/sagemaker/session.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/sagemaker/session.py b/src/sagemaker/session.py index 00797c9ea0..4867509e1b 100644 --- a/src/sagemaker/session.py +++ b/src/sagemaker/session.py @@ -3300,6 +3300,11 @@ def create_endpoint_config_from_existing( if request_data_capture_config_dict is not None: request["DataCaptureConfig"] = request_data_capture_config_dict + if existing_endpoint_config_desc.get("AsyncInferenceConfig") is not None: + request["AsyncInferenceConfig"] = existing_endpoint_config_desc.get( + "AsyncInferenceConfig" + ) + self.sagemaker_client.create_endpoint_config(**request) def create_endpoint(self, endpoint_name, config_name, tags=None, wait=True): From fa91e5ef16afcd1cc8136317f77dfee39c0193f8 Mon Sep 17 00:00:00 2001 From: Shreya Pandit Date: Fri, 20 May 2022 16:55:36 -0700 Subject: [PATCH 2/2] update default value --- src/sagemaker/session.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/sagemaker/session.py b/src/sagemaker/session.py index 4867509e1b..a49c3cd30f 100644 --- a/src/sagemaker/session.py +++ b/src/sagemaker/session.py @@ -3302,7 +3302,7 @@ def create_endpoint_config_from_existing( if existing_endpoint_config_desc.get("AsyncInferenceConfig") is not None: request["AsyncInferenceConfig"] = existing_endpoint_config_desc.get( - "AsyncInferenceConfig" + "AsyncInferenceConfig", None ) self.sagemaker_client.create_endpoint_config(**request)