Skip to content

Commit 1ff8ae3

Browse files
BasilBeiroutiBasil Beirouti
and
Basil Beirouti
authored
fix endpoint bug (#2772)
Co-authored-by: Basil Beirouti <[email protected]>
1 parent 53061e8 commit 1ff8ae3

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)