fix some bug

Author: lt

vllm_ascend/eplb/eplb_updator.py

@@ -1,4 +1,3 @@
-#
 # Copyright (c) 2025 Huawei Technologies Co., Ltd. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -76,7 +75,7 @@ def init_eplb(self, expert_map_path):
             shared_dict = self.shared_dict,
             planner_q = self.planner_block_queue,
             block_update_q = self.block_update_queue,
-            redundant_enable = self.redundant_enable, 
+            redundant_enable = self.redundant_enable,
             policy_type = 6,
             enable_d2d = True
         )
@@ -104,7 +103,6 @@ def wakeup_eplb_worker(self):
         self.planner_block_queue.put(1)
 
     def forward_before(self):
-
         # Batch after eplb process being triggered, get update info provided by eplb process
         if self.update_in_flight and self.weight_update_counter == 0 and self.wait_worker_iterations == self.num_wait_worker_iterations:
             self.wait_worker_iterations = 0

vllm_ascend/worker/worker_v1.py

@@ -178,11 +178,6 @@ def execute_model(
         output = self.model_runner.execute_model(scheduler_output)
         return output if self.is_driver_worker else None
 
-    def get_expert_load(self) -> str:
-        """ todo 一共几个worker"""
-        moe_load = self.model_runner.do_get_expert_load()
-        return moe_load
-
     def load_model(self) -> None:
         if self.vllm_config.model_config.enable_sleep_mode:
             allocator = CaMemAllocator.get_instance()
@@ -215,7 +210,6 @@ def compile_or_warm_up_model(self) -> None:
         set_random_seed(self.model_config.seed)
 
     def get_expert_load(self) -> str:
-        """ todo 一共几个worker"""
         moe_load = self.model_runner.do_get_expert_load()
         return moe_load