Skip to content

Commit 786beb2

Browse files
shreyapanditclaytonparnell
authored andcommitted
Use Async Inference Config when available for endpoint update
1 parent f2d5e41 commit 786beb2

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/sagemaker/session.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3101,6 +3101,11 @@ def create_endpoint_config_from_existing(
31013101
if request_data_capture_config_dict is not None:
31023102
request["DataCaptureConfig"] = request_data_capture_config_dict
31033103

3104+
if existing_endpoint_config_desc.get("AsyncInferenceConfig") is not None:
3105+
request["AsyncInferenceConfig"] = existing_endpoint_config_desc.get(
3106+
"AsyncInferenceConfig"
3107+
)
3108+
31043109
self.sagemaker_client.create_endpoint_config(**request)
31053110

31063111
def create_endpoint(self, endpoint_name, config_name, tags=None, wait=True):

0 commit comments

Comments
 (0)