Skip to content

Commit 61dbafa

Browse files
committed
Use Async Inference Config when available for endpoint update
1 parent 255a339 commit 61dbafa

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
@@ -3055,6 +3055,11 @@ def create_endpoint_config_from_existing(
30553055
if request_data_capture_config_dict is not None:
30563056
request["DataCaptureConfig"] = request_data_capture_config_dict
30573057

3058+
if existing_endpoint_config_desc.get("AsyncInferenceConfig") is not None:
3059+
request["AsyncInferenceConfig"] = existing_endpoint_config_desc.get(
3060+
"AsyncInferenceConfig"
3061+
)
3062+
30583063
self.sagemaker_client.create_endpoint_config(**request)
30593064

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

0 commit comments

Comments
 (0)