"...ssh:/git@developer.sourcefind.cn:2222/OpenDAS/dynamo.git" did not exist on "5f98e91e97488f43242939289d1b2acff8a9ecb3"
Unverified Commit 652c83b6 authored by Antoni Baum's avatar Antoni Baum Committed by GitHub
Browse files

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

parent 6d646d08
...@@ -1671,11 +1671,19 @@ class LLMEngine: ...@@ -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]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment