Rename method to set_inband_engine_stats in sequence group

Signed-off-by: kunjan <kunjanp@google.com>

Author: coolkp

vllm/engine/llm_engine.py

@@ -1166,7 +1166,7 @@ def _process_model_outputs(self,
                 stats_snapshot = self._get_stats(scheduler_outputs, outputs,
                                                  finished_before, skip)
                 inband_stats = self._get_inband_engine_stats(stats_snapshot)
-                seq_group.set_final_engine_stats_snapshot(inband_stats)
+                seq_group.set_inband_engine_stats(inband_stats)
             request_output = RequestOutputFactory.create(
                 seq_group,
                 self.seq_id_to_seq_group,
@@ -1213,7 +1213,7 @@ def _process_model_outputs(self,
                 stats_snapshot = self._get_stats(scheduler_outputs, outputs,
                                                  finished_before, skip)
                 inband_stats = self._get_inband_engine_stats(stats_snapshot)
-                seq_group.set_final_engine_stats_snapshot(inband_stats)
+                seq_group.set_inband_engine_stats(inband_stats)
                 seq_group.set_last_token_time(now)
             request_output = RequestOutputFactory.create(
                 seq_group,

vllm/sequence.py

@@ -848,8 +848,7 @@ def set_finished_time(self, time: Optional[float]) -> None:
         """Sets the finished time for Request level timings."""
         self.metrics.finished_time = time
 
-    def set_final_engine_stats_snapshot(self,
-                                        stats: InbandEngineStats) -> None:
+    def set_inband_engine_stats(self, stats: InbandEngineStats) -> None:
         self.inband_engine_stats = stats
 
     def get_max_num_running_seqs(self) -> int: