fix: add pre_merge markers to vLLM unit tests (#6482)

Signed-off-by: alec-flowers <aflowers@nvidia.com>
Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>

components/src/dynamo/vllm/tests/test_vllm_kv_events_api.py

@@ -32,6 +32,8 @@ KVCacheEvent = _kv_events.KVCacheEvent
 pytestmark = [
     pytest.mark.vllm,
     pytest.mark.unit,
+    pytest.mark.gpu_0,
+    pytest.mark.pre_merge,
 ]
 
 
@@ -49,7 +51,6 @@ class TestVllmKvEventsApi:
         5. lora_id
         6. medium
         7. lora_name (added in vLLM 0.14.0)
-        8. extra_keys (contains MM info, cache_salt, etc.)
 
         If vLLM adds/removes/reorders fields, this test will fail.
         """
@@ -61,7 +62,6 @@ class TestVllmKvEventsApi:
             "lora_id",
             "medium",
             "lora_name",
-            "extra_keys",
         )
 
         actual_fields = BlockStored.__struct_fields__
@@ -148,7 +148,6 @@ class TestVllmKvEventsApi:
             lora_id=None,
             medium="GPU",
             lora_name=None,
-            extra_keys=None,
         )
 
         encoded = msgspec.msgpack.encode(event)
@@ -160,9 +159,9 @@ class TestVllmKvEventsApi:
             decoded[0] == "BlockStored"
         ), f"Expected tag 'BlockStored', got {decoded[0]}"
 
-        # Verify field count (tag + 8 fields = 9 elements)
-        assert len(decoded) == 9, (
-            f"Expected 9 elements (tag + 8 fields), got {len(decoded)}.\n"
+        # Verify field count (tag + 7 fields = 8 elements)
+        assert len(decoded) == 8, (
+            f"Expected 8 elements (tag + 7 fields), got {len(decoded)}.\n"
             f"Decoded: {decoded}\n"
             f"If field count changed, update Rust deserializers."
         )
@@ -175,4 +174,3 @@ class TestVllmKvEventsApi:
         assert decoded[5] is None, f"lora_id at wrong position: {decoded[5]}"
         assert decoded[6] == "GPU", f"medium at wrong position: {decoded[6]}"
         assert decoded[7] is None, f"lora_name at wrong position: {decoded[7]}"
-        assert decoded[8] is None, f"extra_keys at wrong position: {decoded[8]}"

components/src/dynamo/vllm/tests/test_vllm_prompt_embeds.py

@@ -16,6 +16,8 @@ from dynamo.vllm.handlers import BaseWorkerHandler
 pytestmark = [
     pytest.mark.unit,
     pytest.mark.vllm,
+    pytest.mark.gpu_0,
+    pytest.mark.pre_merge,
 ]
 
 

components/src/dynamo/vllm/tests/test_vllm_renderer_api.py

@@ -51,6 +51,8 @@ OutputProcessorOutput = _output_processor_mod.OutputProcessorOutput
 pytestmark = [
     pytest.mark.vllm,
     pytest.mark.unit,
+    pytest.mark.gpu_0,
+    pytest.mark.pre_merge,
 ]