fix annotation

Signed-off-by: youkaichao <[email protected]>
vllm-project · Dec 10, 2024 · 187015f · 187015f
1 parent 16d2a6d
commit 187015f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/vllm/forward_context.py b/vllm/forward_context.py
@@ -10,10 +10,10 @@
 
 logger = init_logger(__name__)
 
-track_batchsize = envs.VLLM_LOG_BATCHSIZE_INTERVAL >= 0
-batchsize_counter = Counter()
+track_batchsize: bool = envs.VLLM_LOG_BATCHSIZE_INTERVAL >= 0
+batchsize_counter: Counter = Counter()
 last_logging_time: float = 0
-batchsize_logging_interval = envs.VLLM_LOG_BATCHSIZE_INTERVAL
+batchsize_logging_interval: float = envs.VLLM_LOG_BATCHSIZE_INTERVAL
 
 
 @dataclass