Skip to content

Commit d889212

Browse files
shreyapanditnavinsoni
authored andcommitted
Use Async Inference Config when available for endpoint update
1 parent 19efadf commit d889212

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
@@ -3300,6 +3300,11 @@ def create_endpoint_config_from_existing(
33003300
if request_data_capture_config_dict is not None:
33013301
request["DataCaptureConfig"] = request_data_capture_config_dict
33023302

3303+
if existing_endpoint_config_desc.get("AsyncInferenceConfig") is not None:
3304+
request["AsyncInferenceConfig"] = existing_endpoint_config_desc.get(
3305+
"AsyncInferenceConfig"
3306+
)
3307+
33033308
self.sagemaker_client.create_endpoint_config(**request)
33043309

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

0 commit comments

Comments
 (0)