diff --git a/vllm/engine/metrics.py b/vllm/engine/metrics.py index 6624837c959c1..ebdbf4dbd9564 100644 --- a/vllm/engine/metrics.py +++ b/vllm/engine/metrics.py @@ -583,7 +583,7 @@ def log(self, stats: Stats) -> None: stats.to_log(self) def log_general_stats(self, stats: GeneralStats): - """Logs GeneralStats to prometheus and tracked stats every iteration.""" + """Logs GeneralStats""" # Log to prometheus. self._log_prometheus_general(stats) @@ -634,7 +634,7 @@ def log_general_stats(self, stats: GeneralStats): self.spec_decode_metrics = None def log_embedding_stats(self, stats: EmbeddingStats): - """Logs EmbeddingStats to prometheus and tracked stats every iteration.""" + """Logs EmbeddingStats""" # Log to prometheus. self._log_prometheus_embedding(stats) diff --git a/vllm/engine/metrics_types.py b/vllm/engine/metrics_types.py index f6b65f04c7425..4966bce687e8e 100644 --- a/vllm/engine/metrics_types.py +++ b/vllm/engine/metrics_types.py @@ -22,7 +22,8 @@ @dataclass class Stats: """Created by LLMEngine for use by StatLogger. - This is the base class for Stats, implementing specified Stats according to needs.""" + This is the base class for Stats, implementing specified Stats + according to needs.""" @abstractmethod def to_log(self, logger: "StatLoggerBase"): @@ -135,5 +136,6 @@ def info(self, type: str, obj: SupportsMetricsInfo) -> None: def maybe_update_spec_decode_metrics(self, stats: Stats): """Save spec decode metrics (since they are unlikely to be emitted at same time as log interval).""" - if isinstance(stats, GeneralStats) and stats.spec_decode_metrics is not None: + if (isinstance(stats, GeneralStats) + and stats.spec_decode_metrics is not None): self.spec_decode_metrics = stats.spec_decode_metrics