Skip to content

Commit 46dcfad

Browse files
committed
Update log to log_logger to distinguish statlogger
1 parent f82e9b5 commit 46dcfad

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/utils/metrics.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -172,12 +172,12 @@ def __init__(self, labels: List[str], max_model_len: int):
172172
class VllmStatLogger(VllmStatLoggerBase):
173173
"""StatLogger is used as an adapter between vLLM stats collector and Triton metrics provider."""
174174

175-
def __init__(self, labels: Dict, max_model_len: int, logger) -> None:
175+
def __init__(self, labels: Dict, max_model_len: int, log_logger) -> None:
176176
# Tracked stats over current local logging interval.
177177
# local_interval not used here. It's for vLLM logs to stdout.
178178
super().__init__(local_interval=0)
179179
self.metrics = TritonMetrics(labels, max_model_len)
180-
self.logger = logger
180+
self.log_logger = log_logger
181181

182182
# Starting the metrics thread. It allows vLLM to keep making progress
183183
# while reporting metrics to triton metrics service.
@@ -268,7 +268,7 @@ def logger_loop(self):
268268
elif command == "observe":
269269
metric.observe(data)
270270
else:
271-
self.logger.log_error(f"Undefined command name: {command}")
271+
self.log_logger.log_error(f"Undefined command name: {command}")
272272

273273
def finalize(self):
274274
# Shutdown the logger thread.

0 commit comments

Comments
 (0)