fix lint

Signed-off-by: SlightwindSec <[email protected]>

Author: SlightwindSec

vllm_ascend/distributed/mooncake/mooncake_store_connector_v1.py

@@ -45,7 +45,7 @@ def __init__(self, vllm_config: VllmConfig, role: KVConnectorRole):
             )
 
             assert self.connector_worker is not None
-            if vllm_config.parallel_config.rank == 0 and self.kv_role == "kv_producer":
+            if vllm_config.parallel_config.rank == 0 and self.kv_role != "kv_consumer":
                 self.lookup_server = MooncakeLookupServer(
                     self.connector_worker, vllm_config, self.use_layerwise)
 
@@ -163,7 +163,7 @@ def __init__(self, vllm_config: "VllmConfig", use_layerwise):
         self.use_layerwise = use_layerwise
         self.kv_role = vllm_config.kv_transfer_config.kv_role
         self.client = MooncakeLookupClient(
-            vllm_config) if self.kv_role == "kv_producer" else None
+            vllm_config) if self.kv_role != "kv_consumer" else None
         self.consumer_is_to_load = vllm_config.kv_transfer_config.kv_connector_extra_config.get(
             "consumer_is_to_load", False)
         self.load_async = vllm_config.kv_transfer_config.kv_connector_extra_config.get(
@@ -208,7 +208,7 @@ def get_num_new_matched_tokens(
         else:
             token_ids = torch.tensor(request.prompt_token_ids)
 
-        num_external_hit_tokens = self.client.lookup(token_ids)
+        num_external_hit_tokens = self.client.lookup(token_ids)  # type: ignore[union-attr]
 
         if num_external_hit_tokens == request.num_tokens:
             num_external_hit_tokens -= 1