Fix circular import in test_prefix_chunk_info.py (#7097)

25a6a9aa · Baizhou Zhang · GitHub · 83d87685 · 25a6a9aa
Unverified Commit 25a6a9aa authored Jun 11, 2025 by Baizhou Zhang Committed by GitHub Jun 11, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

python/sglang/test/attention/test_prefix_chunk_info.py python/sglang/test/attention/test_prefix_chunk_info.py +2 -0

No files found.
--- 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