Skip to content

Commit bfad142

Browse files
authored
[BUGFIX][NIXL] quick fix for 'assert self.connector_worker is not None' in get_kv_connector_stats (#26851)
Signed-off-by: Chendi Xue <chendi.xue@intel.com>
1 parent 9354660 commit bfad142

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

vllm/distributed/kv_transfer/kv_connector/v1/nixl_connector.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -241,7 +241,8 @@ def get_block_ids_with_load_errors(self) -> set[int]:
241241
return self.connector_worker.get_block_ids_with_load_errors()
242242

243243
def get_kv_connector_stats(self) -> KVConnectorStats | None:
244-
assert self.connector_worker is not None
244+
if self.connector_worker is None:
245+
return None
245246
return self.connector_worker.get_kv_connector_stats()
246247

247248
@classmethod

0 commit comments

Comments
 (0)