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

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged

Feb 20, 2026

-Original file line number
+Diff line change
@@ Expand Up / @@ -17,6 +17,7 @@ @@
         ModelConfig,
         SchedulerConfig,
         VllmConfig,
+        set_current_vllm_config,
     )
     from vllm.distributed.kv_transfer.kv_connector.v1.moriio.moriio_common import (
         MoRIIOAgentMetadata,
@@ Expand Down Expand Up / @@ -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,
@@ Expand Down Expand Up @@
                     "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)
@@ Expand Down @@

Provide feedback