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/run_suite.py
+++ b/test/srt/run_suite.py
@@ -72,6 +72,7 @@ suites = {
        TestFile("test_vision_openai_server.py", 344),
        TestFile("test_w8a8_quantization.py", 46),
        TestFile("test_eval_fp8_accuracy.py", 172),
+        TestFile("test_create_kvindices.py", 2),
    ],
    "nightly": [
        TestFile("test_nightly_gsm8k_eval.py"),
--- 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):