Skip to content

GitLab

Unverified Commit 25a6a9aa authored by Baizhou Zhang's avatar Baizhou Zhang Committed by GitHub
Browse files

Fix circular import in test_prefix_chunk_info.py (#7097)

parent 83d87685
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
python/sglang/test/attention/test_prefix_chunk_info.py
View file @ 25a6a9aa
...@@ -2,6 +2,8 @@ import unittest ...@@ -2,6 +2,8 @@ import unittest
import torch 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.mem_cache.memory_pool import MLATokenToKVPool
from sglang.srt.model_executor.forward_batch_info import ForwardBatch, ForwardMode from sglang.srt.model_executor.forward_batch_info import ForwardBatch, ForwardMode
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment