Fix circular import in test_prefix_chunk_info.py (#7097)

2025-06-11 10:57:45 -07:00
parent 83d87685c5
commit 25a6a9aa22
1 changed files with 2 additions and 0 deletions
--- 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