Skip to content

Commit 507db15

Browse files
author
Pravali Uppugunduri
committed
Trigger PR checks
1 parent 5901a6a commit 507db15

File tree

2 files changed

+1
-3
lines changed

2 files changed

+1
-3
lines changed

src/sagemaker/serve/builder/jumpstart_builder.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,6 @@ def __init__(self):
121121
self.is_compiled = False
122122
self.is_quantized = False
123123
self.speculative_decoding_draft_model_source = None
124-
self.is_set_deployment_config = False
125124
self.deployment_config_name = None
126125

127126
@abstractmethod
@@ -513,7 +512,6 @@ def set_deployment_config(self, config_name: str, instance_type: str) -> None:
513512
raise Exception("Cannot set deployment config to an uninitialized model.")
514513

515514
self.pysdk_model.set_deployment_config(config_name, instance_type)
516-
is_set_deployment_config = True
517515
self.deployment_config_name = config_name
518516

519517
self.instance_type = instance_type

src/sagemaker/serve/utils/telemetry_logger.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -164,7 +164,7 @@ def wrapper(self, *args, **kwargs):
164164
)
165165
model_provider_value = SD_DRAFT_MODEL_SOURCE_TO_CODE[str(model_provider_enum)]
166166
extra += f"&x-sdDraftModelSource={model_provider_value}"
167-
167+
168168
if getattr(self, "deployment_config_name", False):
169169
config_name_code = self.deployment_config_name.lower()
170170
extra += f"&x-configName={config_name_code}"

0 commit comments

Comments
 (0)