Tiny fix wrong metric collect key name forward_prefill -> forward_extend (#18506)

2026-02-09 16:23:41 -08:00
parent 26a006e47f
commit 2825f5d8e0
1 changed files with 5 additions and 2 deletions
--- a/test/registered/metrics/test_metrics.py
+++ b/test/registered/metrics/test_metrics.py
@@ -46,10 +46,13 @@ class TestEnableMetrics(CustomTestCase):
                    "sglang:dp_cooperation_realtime_tokens_total",
                    {"mode": "prefill_compute"},
                ),
-                ("sglang:dp_cooperation_realtime_tokens_total", {"mode": "decode"}),
+                (
+                    "sglang:dp_cooperation_realtime_tokens_total",
+                    {"mode": "decode"},
+                ),
                (
                    "sglang:dp_cooperation_gpu_execution_seconds_total",
-                    {"category": "forward_prefill"},
+                    {"category": "forward_extend"},
                ),
                (
                    "sglang:dp_cooperation_gpu_execution_seconds_total",