Skip to content

Commit e5be549

Browse files
committed
fix bugs of pre-commit
Signed-off-by: bhsueh <[email protected]>
1 parent e87f92c commit e5be549

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

tests/integration/defs/accuracy/test_llm_api_pytorch.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2736,7 +2736,8 @@ def test_w4_1gpu(self, moe_backend, cuda_graph, overlap_scheduler, mocker):
27362736
with llm:
27372737
model_name = "GPT-OSS/MXFP4"
27382738
mocker.patch.object(GSM8K, "MAX_OUTPUT_LEN", 8192)
2739-
mocker.patch.dict(GSM8K.EVALUATE_KWARGS, {"scores_filter": "exact_match,flexible-extract"})
2739+
mocker.patch.dict(GSM8K.EVALUATE_KWARGS,
2740+
{"scores_filter": "exact_match,flexible-extract"})
27402741
task = GSM8K(model_name)
27412742
task.evaluate(llm,
27422743
extra_evaluator_kwargs=self.extra_evaluator_kwargs)
@@ -2778,7 +2779,8 @@ def test_w4_4gpus(self, moe_backend, tp_size, pp_size, ep_size,
27782779
model_name = "GPT-OSS/MXFP4"
27792780
task = GSM8K(model_name)
27802781
mocker.patch.object(GSM8K, "MAX_OUTPUT_LEN", 8192)
2781-
mocker.patch.dict(GSM8K.EVALUATE_KWARGS, {"scores_filter": "exact_match,flexible-extract"})
2782+
mocker.patch.dict(GSM8K.EVALUATE_KWARGS,
2783+
{"scores_filter": "exact_match,flexible-extract"})
27822784
task.evaluate(llm,
27832785
extra_evaluator_kwargs=self.extra_evaluator_kwargs)
27842786

@@ -2810,7 +2812,8 @@ def test_w4a16(self, tp_size, pp_size, ep_size, attention_dp, cuda_graph,
28102812
model_name = "GPT-OSS/BF16"
28112813
task = GSM8K(model_name)
28122814
mocker.patch.object(GSM8K, "MAX_OUTPUT_LEN", 8192)
2813-
mocker.patch.dict(GSM8K.EVALUATE_KWARGS, {"scores_filter": "exact_match,flexible-extract"})
2815+
mocker.patch.dict(GSM8K.EVALUATE_KWARGS,
2816+
{"scores_filter": "exact_match,flexible-extract"})
28142817
task.evaluate(llm,
28152818
extra_evaluator_kwargs=self.extra_evaluator_kwargs)
28162819

0 commit comments

Comments
 (0)