Skip to content

Commit def2102

Browse files
shreyapanditnavinsoni
authored andcommitted
Use Async Inference Config when available for endpoint update
1 parent a35a093 commit def2102

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
@@ -3128,6 +3128,11 @@ def create_endpoint_config_from_existing(
31283128
if request_data_capture_config_dict is not None:
31293129
request["DataCaptureConfig"] = request_data_capture_config_dict
31303130

3131+
if existing_endpoint_config_desc.get("AsyncInferenceConfig") is not None:
3132+
request["AsyncInferenceConfig"] = existing_endpoint_config_desc.get(
3133+
"AsyncInferenceConfig"
3134+
)
3135+
31313136
self.sagemaker_client.create_endpoint_config(**request)
31323137

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

0 commit comments

Comments
 (0)