Update src/utils/metrics.py

Pavloveuge · yinggeh · web-flow · commit c5f97514788f · 2025-07-24T20:47:01.000+03:00
Co-authored-by: Yingge He &lt;157551214+yinggeh@users.noreply.github.com&gt;
diff --git a/src/utils/metrics.py b/src/utils/metrics.py
@@ -363,7 +363,7 @@ def log(self, stats: VllmStats) -> None:
         counter_metrics = [
             (self.metrics.counter_prompt_tokens, stats.num_prompt_tokens_iter),
             (self.metrics.counter_generation_tokens, stats.num_generation_tokens_iter),
-            (self.metrics.counter_preemption_tokens, stats.num_preemption_iter),
+            (self.metrics.counter_num_preemption, stats.num_preemption_iter),
         ]
         histogram_metrics = [
             (self.metrics.histogram_iteration_tokens, [stats.num_tokens_iter]),

Original file line number	Diff line number	Diff line change
`@@ -363,7 +363,7 @@ def log(self, stats: VllmStats) -> None:`
`363`	`363`	`counter_metrics = [`
`364`	`364`	`(self.metrics.counter_prompt_tokens, stats.num_prompt_tokens_iter),`
`365`	`365`	`(self.metrics.counter_generation_tokens, stats.num_generation_tokens_iter),`
`366`		`- (self.metrics.counter_preemption_tokens, stats.num_preemption_iter),`
	`366`	`+ (self.metrics.counter_num_preemption, stats.num_preemption_iter),`
`367`	`367`	`]`
`368`	`368`	`histogram_metrics = [`
`369`	`369`	`(self.metrics.histogram_iteration_tokens, [stats.num_tokens_iter]),`