From 53adb3cb4e05a852a73bb97b9f870cfc6d522a68 Mon Sep 17 00:00:00 2001 From: Yuan Tong Date: Tue, 25 Mar 2025 18:41:30 +0800 Subject: [PATCH] test: waive flaky test_kv_cache_event_async_api (#3062) Signed-off-by: Yuan Tong <13075180+tongyuantongyu@users.noreply.github.com> --- tests/unittest/llmapi/test_llm_kv_cache_events.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/unittest/llmapi/test_llm_kv_cache_events.py b/tests/unittest/llmapi/test_llm_kv_cache_events.py index 81d5728fd0..85a7feac7f 100644 --- a/tests/unittest/llmapi/test_llm_kv_cache_events.py +++ b/tests/unittest/llmapi/test_llm_kv_cache_events.py @@ -1,6 +1,7 @@ import asyncio import time +import pytest from test_llm import get_model_path import tensorrt_llm @@ -119,6 +120,7 @@ def test_expected_kv_cache_events(): assert event[0]["data"]["type"] == "stored" +@pytest.mark.skip("https://nvbugs/5150466: flaky fail") def test_kv_cache_event_async_api(): llm = create_llm() sampling_params = SamplingParams(max_tokens=6, temperature=0.01)