From f4633fe381018d086741a32779c182c511cd59ec Mon Sep 17 00:00:00 2001 From: Syed Muhammad Bin Asif <92625830+syedmba@users.noreply.github.com> Date: Fri, 28 Nov 2025 15:58:16 +0800 Subject: [PATCH] [Bugfix] Fix removal of old logs when stats are enabled (#84) Signed-off-by: syedmba --- vllm_omni/entrypoints/async_omni_llm.py | 2 +- vllm_omni/entrypoints/omni_llm.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/vllm_omni/entrypoints/async_omni_llm.py b/vllm_omni/entrypoints/async_omni_llm.py index 7cc138efb..730001173 100644 --- a/vllm_omni/entrypoints/async_omni_llm.py +++ b/vllm_omni/entrypoints/async_omni_llm.py @@ -110,7 +110,7 @@ class AsyncOmniLLM(EngineClient): # Optional file handler for orchestrator self._log_file = log_file if self._log_file: - remove_old_logs(self._log_file, len(self.stage_list)) + remove_old_logs(self._log_file, len(self.stage_configs)) configure_orchestrator_logger(logger, self._log_file) self._stats_file, self._overall_stats_file = init_stats_paths(self._enable_stats, self._log_file) diff --git a/vllm_omni/entrypoints/omni_llm.py b/vllm_omni/entrypoints/omni_llm.py index c03d3ce2f..dccc380ae 100644 --- a/vllm_omni/entrypoints/omni_llm.py +++ b/vllm_omni/entrypoints/omni_llm.py @@ -97,7 +97,7 @@ class OmniLLM: # Optional file handler for orchestrator self._log_file = log_file if self._log_file: - remove_old_logs(self._log_file, len(self.stage_list)) + remove_old_logs(self._log_file, len(self.stage_configs)) configure_orchestrator_logger(logger, self._log_file) self._stats_file, self._overall_stats_file = init_stats_paths(self._enable_stats, self._log_file)