Refactor KV connector output emptiness check

Signed-off-by: David Ben-David <[email protected]>

Author: David Ben-David

vllm/v1/worker/gpu_worker.py

@@ -464,9 +464,7 @@ def execute_model(
 
         # In case of PP with kv transfer, we need to pass through the
         # kv_connector_output
-        if (not kv_connector_output.finished_sending
-                and not kv_connector_output.finished_recving
-                and not kv_connector_output.invalid_block_ids):
+        if kv_connector_output.is_empty():
             return EMPTY_MODEL_RUNNER_OUTPUT
 
         output = copy.copy(EMPTY_MODEL_RUNNER_OUTPUT)

vllm/v1/worker/kv_connector_model_runner_mixin.py

@@ -75,9 +75,7 @@ def kv_connector_no_forward(scheduler_output: "SchedulerOutput",
                 scheduler_output, wait_for_save=False) as kv_connector_output:
             pass
 
-        if (not kv_connector_output.finished_sending
-                and not kv_connector_output.finished_recving
-                and not kv_connector_output.invalid_block_ids):
+        if kv_connector_output.is_empty():
             return EMPTY_MODEL_RUNNER_OUTPUT
 
         output = copy.copy(EMPTY_MODEL_RUNNER_OUTPUT)