Skip to content

Commit c35da48

Browse files
author
Basil Beirouti
committed
fix endpoint bug
1 parent 331f0c7 commit c35da48

File tree

1 file changed

+11
-13
lines changed

1 file changed

+11
-13
lines changed

src/sagemaker/session.py

+11-13
Original file line numberDiff line numberDiff line change
@@ -3556,19 +3556,17 @@ def endpoint_from_production_variants(
35563556
Returns:
35573557
str: The name of the created ``Endpoint``.
35583558
"""
3559-
if not _deployment_entity_exists(
3560-
lambda: self.sagemaker_client.describe_endpoint_config(EndpointConfigName=name)
3561-
):
3562-
config_options = {"EndpointConfigName": name, "ProductionVariants": production_variants}
3563-
tags = _append_project_tags(tags)
3564-
if tags:
3565-
config_options["Tags"] = tags
3566-
if kms_key:
3567-
config_options["KmsKeyId"] = kms_key
3568-
if data_capture_config_dict is not None:
3569-
config_options["DataCaptureConfig"] = data_capture_config_dict
3570-
3571-
self.sagemaker_client.create_endpoint_config(**config_options)
3559+
config_options = {"EndpointConfigName": name, "ProductionVariants": production_variants}
3560+
tags = _append_project_tags(tags)
3561+
if tags:
3562+
config_options["Tags"] = tags
3563+
if kms_key:
3564+
config_options["KmsKeyId"] = kms_key
3565+
if data_capture_config_dict is not None:
3566+
config_options["DataCaptureConfig"] = data_capture_config_dict
3567+
3568+
self.sagemaker_client.create_endpoint_config(**config_options)
3569+
35723570
return self.create_endpoint(endpoint_name=name, config_name=name, tags=tags, wait=wait)
35733571

35743572
def expand_role(self, role):

0 commit comments

Comments
 (0)