瀏覽代碼

fix: raise exception when accessing logger for disable_log_stats=True case (#984)

AlpinDale 2 月之前
父節點
當前提交
55fa2b9c54
共有 1 個文件被更改,包括 8 次插入0 次删除
  1. 8 0
      aphrodite/engine/aphrodite_engine.py

+ 8 - 0
aphrodite/engine/aphrodite_engine.py

@@ -1580,11 +1580,19 @@ class AphroditeEngine:
         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]