From 9a628744fcefb0aaf4920cb673ffee7f1d93458c Mon Sep 17 00:00:00 2001 From: khalilzhk Date: Mon, 12 Jan 2026 20:16:30 +0800 Subject: [PATCH] [CI] fix piecewise graph test case on ascend (#16933) --- test/srt/ascend/test_ascend_piecewise_graph_prefill.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/test/srt/ascend/test_ascend_piecewise_graph_prefill.py b/test/srt/ascend/test_ascend_piecewise_graph_prefill.py index 9e43ca60f..83f094df2 100644 --- a/test/srt/ascend/test_ascend_piecewise_graph_prefill.py +++ b/test/srt/ascend/test_ascend_piecewise_graph_prefill.py @@ -38,8 +38,8 @@ class TestPiecewiseGraphPrefillCorrectness(CustomTestCase): 128, "--enable-piecewise-cuda-graph", "--piecewise-cuda-graph-tokens", - TOKENS_TO_CAPTURE, - ], + ] + + TOKENS_TO_CAPTURE, ) @classmethod @@ -79,8 +79,8 @@ class TestPiecewiseGraphPrefillBenchmark(CustomTestCase): "ascend", "--enable-piecewise-cuda-graph", "--piecewise-cuda-graph-tokens", - TOKENS_TO_CAPTURE, - ], + ] + + TOKENS_TO_CAPTURE, ) self.assertLess(prefill_latency, EXP_PREFILL_LATENCY)