From 0c1dc42748760fc75aef68e973c9ff7a47501337 Mon Sep 17 00:00:00 2001 From: rasmith Date: Thu, 19 Feb 2026 23:32:40 -0600 Subject: [PATCH] [CI][AMD][BugFix][P/D] Add default_vllm_config to test_moriio_connector.py so tests pass (#33739) Signed-off-by: Randall Smith --- tests/v1/kv_connector/unit/test_moriio_connector.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/tests/v1/kv_connector/unit/test_moriio_connector.py b/tests/v1/kv_connector/unit/test_moriio_connector.py index 1cc698863..1eca4964f 100644 --- a/tests/v1/kv_connector/unit/test_moriio_connector.py +++ b/tests/v1/kv_connector/unit/test_moriio_connector.py @@ -17,6 +17,7 @@ from vllm.config import ( ModelConfig, SchedulerConfig, VllmConfig, + set_current_vllm_config, ) from vllm.distributed.kv_transfer.kv_connector.v1.moriio.moriio_common import ( MoRIIOAgentMetadata, @@ -433,10 +434,11 @@ def test_register_kv_caches(mock_parallel_groups): } ) - connector = MoRIIOConnector(vllm_config, KVConnectorRole.WORKER) - connector.connector_worker = FakeMorIIOConnectorWorker( - vllm_config, connector.engine_id, hand_shake_latency=0 - ) + with set_current_vllm_config(vllm_config): + connector = MoRIIOConnector(vllm_config, KVConnectorRole.WORKER) + connector.connector_worker = FakeMorIIOConnectorWorker( + vllm_config, connector.engine_id, hand_shake_latency=0 + ) from mori.io import ( MemoryDesc, @@ -523,7 +525,8 @@ def test_moriio_handshake_returns_metadata(mock_parallel_groups): "handshake_port": handshake_port, } ) - connector = MoRIIOConnector(vllm_config, KVConnectorRole.WORKER) + with set_current_vllm_config(vllm_config): + connector = MoRIIOConnector(vllm_config, KVConnectorRole.WORKER) # Execute register_kv_caches connector.register_kv_caches(kv_caches)