Skip to content
This repository was archived by the owner on Sep 4, 2025. It is now read-only.

Commit 652c83b

Browse files
authored
[Misc] Raise a more informative exception in add/remove_logger (vllm-project#7750)
1 parent 6d646d0 commit 652c83b

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

vllm/engine/llm_engine.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1671,11 +1671,19 @@ def _get_last_sampled_token_ids(
16711671
return None
16721672

16731673
def add_logger(self, logger_name: str, logger: StatLoggerBase) -> None:
1674+
if not self.log_stats:
1675+
raise RuntimeError(
1676+
"Stat logging is disabled. Set `disable_log_stats=False` "
1677+
"argument to enable.")
16741678
if logger_name in self.stat_loggers:
16751679
raise KeyError(f"Logger with name {logger_name} already exists.")
16761680
self.stat_loggers[logger_name] = logger
16771681

16781682
def remove_logger(self, logger_name: str) -> None:
1683+
if not self.log_stats:
1684+
raise RuntimeError(
1685+
"Stat logging is disabled. Set `disable_log_stats=False` "
1686+
"argument to enable.")
16791687
if logger_name not in self.stat_loggers:
16801688
raise KeyError(f"Logger with name {logger_name} does not exist.")
16811689
del self.stat_loggers[logger_name]

0 commit comments

Comments
 (0)