Unverified Commit 0c1dc427 authored by rasmith's avatar rasmith Committed by GitHub
Browse files

[CI][AMD][BugFix][P/D] Add default_vllm_config to test_moriio_connector.py so tests pass (#33739)


Signed-off-by: default avatarRandall Smith <Randall.Smith@amd.com>
parent 676f82ae
...@@ -17,6 +17,7 @@ from vllm.config import ( ...@@ -17,6 +17,7 @@ from vllm.config import (
ModelConfig, ModelConfig,
SchedulerConfig, SchedulerConfig,
VllmConfig, VllmConfig,
set_current_vllm_config,
) )
from vllm.distributed.kv_transfer.kv_connector.v1.moriio.moriio_common import ( from vllm.distributed.kv_transfer.kv_connector.v1.moriio.moriio_common import (
MoRIIOAgentMetadata, MoRIIOAgentMetadata,
...@@ -433,10 +434,11 @@ def test_register_kv_caches(mock_parallel_groups): ...@@ -433,10 +434,11 @@ def test_register_kv_caches(mock_parallel_groups):
} }
) )
connector = MoRIIOConnector(vllm_config, KVConnectorRole.WORKER) with set_current_vllm_config(vllm_config):
connector.connector_worker = FakeMorIIOConnectorWorker( connector = MoRIIOConnector(vllm_config, KVConnectorRole.WORKER)
vllm_config, connector.engine_id, hand_shake_latency=0 connector.connector_worker = FakeMorIIOConnectorWorker(
) vllm_config, connector.engine_id, hand_shake_latency=0
)
from mori.io import ( from mori.io import (
MemoryDesc, MemoryDesc,
...@@ -523,7 +525,8 @@ def test_moriio_handshake_returns_metadata(mock_parallel_groups): ...@@ -523,7 +525,8 @@ def test_moriio_handshake_returns_metadata(mock_parallel_groups):
"handshake_port": handshake_port, "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 # Execute register_kv_caches
connector.register_kv_caches(kv_caches) connector.register_kv_caches(kv_caches)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment