Skip to content

Commit 2665231

Browse files
committed
update formatting
1 parent 06ed0e3 commit 2665231

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/sagemaker/serve/validations/optimization.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ def validate_against(self, optimization_combination, rule_set: _OptimizationCont
6666
is_compiled = optimization_combination.compilation.copy().pop()
6767
is_quantized = optimization_combination.quantization_technique.copy().pop()
6868
if is_quantized and not is_compiled:
69-
raise ValueError(f"Quantization must be provided with Compilation")
69+
raise ValueError("Quantization must be provided with Compilation")
7070

7171

7272
TRUTHY_SET = {None, True}

tests/unit/sagemaker/serve/builder/test_model_builder.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2933,7 +2933,7 @@ def test_trt_configurations_rule_set(self):
29332933
_validate_optimization_configuration(
29342934
instance_type="ml.g5.24xlarge",
29352935
quantization_config={
2936-
"OverrideEnvironment": {"OPTION_QUANTIZE": "awq"},
2936+
"OverrideEnvironment": {"OPTION_QUANTIZE": "smoothquant"},
29372937
},
29382938
sharding_config=None,
29392939
speculative_decoding_config=None,
@@ -2959,7 +2959,7 @@ def test_vllm_configurations_rule_set(self):
29592959
compilation_config=None,
29602960
)
29612961

2962-
# Cab be quantized
2962+
# Can be quantized
29632963
_validate_optimization_configuration(
29642964
instance_type="ml.g5.24xlarge",
29652965
quantization_config={

0 commit comments

Comments
 (0)