Fix test_create_kvindices unit test (#4452)

2025-03-16 07:01:04 +08:00
parent 035ac2ab74
commit 21d485f835
2 changed files with 2 additions and 3 deletions
--- a/test/srt/test_create_kvindices.py
+++ b/test/srt/test_create_kvindices.py
@@ -4,9 +4,7 @@ import unittest
 import numpy as np
 import torch

-from sglang.srt.layers.attention.flashinfer_utils import (
-    create_flashinfer_kv_indices_triton,
-)
+from sglang.srt.layers.attention.utils import create_flashinfer_kv_indices_triton


 class TestCreateKvIndices(unittest.TestCase):