kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-14 06:27:45 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

test: waive flaky test_kv_cache_event_async_api (#3062)

Browse Source 
Signed-off-by: Yuan Tong <13075180+tongyuantongyu@users.noreply.github.com>
This commit is contained in:
Yuan Tong 2025-03-25 18:41:30 +08:00 committed by GitHub
parent d9acce72bb
commit 53adb3cb4e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/unittest/llmapi/test_llm_kv_cache_events.py
View File

@ -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)