diff --git a/components/src/dynamo/vllm/publisher.py b/components/src/dynamo/vllm/publisher.py index a3d13d7049..3e69feec98 100644 --- a/components/src/dynamo/vllm/publisher.py +++ b/components/src/dynamo/vllm/publisher.py @@ -63,6 +63,8 @@ def record( scheduler_stats: SchedulerStats, iteration_stats: Optional[IterationStats], engine_idx: int = 0, + *args, + **kwargs, ): # request_total_slots and kv_total_blocks are properties of model + gpu # we should only publish them once, not every metric update