Browse Source

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

AlpinDale 2 months ago
parent
commit
55fa2b9c54
1 changed files with 8 additions and 0 deletions
  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]