Remove unnecessary add callback

AlibabaPAI · Aug 29, 2024 · c5adfa8 · c5adfa8
1 parent 5f3286d
commit c5adfa8
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/tests/backends/vllm/test_llm_engine.py b/tests/backends/vllm/test_llm_engine.py
@@ -31,7 +31,6 @@
 class MockEngine(LLMEngineLlumnix):
     def __init__(self, executor_class=None, *args, **kwargs):
         self.scheduler = initialize_scheduler()
-        self.scheduler.add_update_instance_info_callback(self.update_instance_info)
         detokenizer = MagicMock(spec=Detokenizer)
         stop_checker = MagicMock(spec=StopChecker)
         seq_counter = Counter()