From 9db769ee6280748420f6d66628c5aaf311c85362 Mon Sep 17 00:00:00 2001 From: Emma Qiao Date: Sun, 29 Jun 2025 11:06:14 +0800 Subject: [PATCH] [Infra] - Add import pytest (#5565) Signed-off-by: qqiao --- 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 691c394c55..c21d12bd3a 100644 --- a/tests/unittest/llmapi/test_llm_kv_cache_events.py +++ b/tests/unittest/llmapi/test_llm_kv_cache_events.py @@ -1,6 +1,8 @@ import asyncio import time +import pytest + import tensorrt_llm from tensorrt_llm._tensorrt_engine import LLM from tensorrt_llm._torch.pyexecutor.llm_request import LlmRequest