[Misc] Raise a more informative exception in add/remove_logger (#7750)
This commit is contained in:
parent
6d646d08a2
commit
652c83b697
@ -1671,11 +1671,19 @@ class LLMEngine:
|
||||
return 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:
|
||||
raise KeyError(f"Logger with name {logger_name} already exists.")
|
||||
self.stat_loggers[logger_name] = logger
|
||||
|
||||
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:
|
||||
raise KeyError(f"Logger with name {logger_name} does not exist.")
|
||||
del self.stat_loggers[logger_name]
|
||||
|
||||
Loading…
Reference in New Issue
Block a user