Skip to content

Commit 2523e46

Browse files
shreyapanditnavinsoni
authored andcommitted
Use Async Inference Config when available for endpoint update (#3124)
Co-authored-by: Navin Soni <[email protected]>
1 parent 92358fe commit 2523e46

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
@@ -3324,6 +3324,11 @@ def create_endpoint_config_from_existing(
33243324
if request_data_capture_config_dict is not None:
33253325
request["DataCaptureConfig"] = request_data_capture_config_dict
33263326

3327+
if existing_endpoint_config_desc.get("AsyncInferenceConfig") is not None:
3328+
request["AsyncInferenceConfig"] = existing_endpoint_config_desc.get(
3329+
"AsyncInferenceConfig", None
3330+
)
3331+
33273332
self.sagemaker_client.create_endpoint_config(**request)
33283333

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

0 commit comments

Comments
 (0)