From 25a6a9aa226c3063f76832d4644fd0bbffe03454 Mon Sep 17 00:00:00 2001 From: Baizhou Zhang Date: Wed, 11 Jun 2025 10:57:45 -0700 Subject: [PATCH] Fix circular import in test_prefix_chunk_info.py (#7097) --- python/sglang/test/attention/test_prefix_chunk_info.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/sglang/test/attention/test_prefix_chunk_info.py b/python/sglang/test/attention/test_prefix_chunk_info.py index 2b85b695b..c02d4d1d6 100644 --- a/python/sglang/test/attention/test_prefix_chunk_info.py +++ b/python/sglang/test/attention/test_prefix_chunk_info.py @@ -2,6 +2,8 @@ import unittest import torch +from sglang.srt.layers.attention.flashattention_backend import FlashAttentionBackend +from sglang.srt.layers.radix_attention import RadixAttention from sglang.srt.mem_cache.memory_pool import MLATokenToKVPool from sglang.srt.model_executor.forward_batch_info import ForwardBatch, ForwardMode from sglang.test.test_utils import CustomTestCase