🐛 fix assert

Signed-off-by: Prashant Gupta <[email protected]>
opendatahub-io · Jul 1, 2024 · afe4de5 · afe4de5
1 parent 595a317
commit afe4de5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/vllm_tgis_adapter/grpc/grpc_server.py b/src/vllm_tgis_adapter/grpc/grpc_server.py
@@ -191,8 +191,8 @@ async def post_init(self) -> None:
         assert self.tokenizer is not None
 
         # Swap in the special TGIS stats logger
-        assert hasattr(self.engine.engine, "stat_logger")
-        assert self.engine.engine.stat_logger
+        assert hasattr(self.engine.engine, "stat_loggers")
+        assert self.engine.engine.stat_loggers
 
         vllm_stat_logger = self.engine.engine.stat_loggers["prometheus"]
         tgis_stats_logger = TGISStatLogger(