squall/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[Misc] Raise a more informative exception in add/remove_logger (#7750)

Browse Source
This commit is contained in:
Antoni Baum 2024-09-03 12:28:25 -07:00 committed by GitHub
parent 6d646d08a2
commit 652c83b697
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
vllm/engine/llm_engine.py
View File

@ -1671,11 +1671,19 @@ class LLMEngine:
return None return None
def add_logger(self, logger_name: str, logger: StatLoggerBase) -> None: def add_logger(self, logger_name: str, logger: StatLoggerBase) -> None:
if not self.log_stats:
raise RuntimeError(
"Stat logging is disabled. Set `disable_log_stats=False` "
"argument to enable.")
if logger_name in self.stat_loggers: if logger_name in self.stat_loggers:
raise KeyError(f"Logger with name {logger_name} already exists.") raise KeyError(f"Logger with name {logger_name} already exists.")
self.stat_loggers[logger_name] = logger self.stat_loggers[logger_name] = logger
def remove_logger(self, logger_name: str) -> None: def remove_logger(self, logger_name: str) -> None:
if not self.log_stats:
raise RuntimeError(
"Stat logging is disabled. Set `disable_log_stats=False` "
"argument to enable.")
if logger_name not in self.stat_loggers: if logger_name not in self.stat_loggers:
raise KeyError(f"Logger with name {logger_name} does not exist.") raise KeyError(f"Logger with name {logger_name} does not exist.")
del self.stat_loggers[logger_name] del self.stat_loggers[logger_name]