diff --git a/tests/v1/kv_connector/unit/test_nixl_desc_geometry.py b/tests/v1/kv_connector/unit/test_nixl_desc_geometry.py index 607e4822b9..bfec51da69 100644 --- a/tests/v1/kv_connector/unit/test_nixl_desc_geometry.py +++ b/tests/v1/kv_connector/unit/test_nixl_desc_geometry.py @@ -156,6 +156,8 @@ def test_overlaid_transfer_groups_share_region_geometry(): worker = object.__new__(NixlConnectorWorker) worker.tp_rank = 0 worker.world_size = 1 + worker.transfer_tp_rank = 0 + worker.transfer_tp_size = 1 worker.block_size = 4 worker.engine_id = "local-engine" worker.use_mla = True