fix after rebase

sgl-project · Aug 26, 2024 · 00fd8b4 · 00fd8b4
1 parent 1d17d6d
commit 00fd8b4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/python/sglang/srt/managers/tp_worker.py b/python/sglang/srt/managers/tp_worker.py
@@ -28,7 +28,7 @@
 import torch.distributed as dist
 
 from sglang.global_config import global_config
-from sglang.srt.constrained.fsm_cache import FSMCache, FSMJsonCache
+from sglang.srt.constrained.fsm_cache import FSMCache
 from sglang.srt.constrained.jump_forward import JumpForwardCache
 from sglang.srt.hf_transformers_utils import get_processor, get_tokenizer
 from sglang.srt.layers.logits_processor import LogitsProcessorOutput

diff --git a/test/srt/test_json_constrained.py b/test/srt/test_json_constrained.py
@@ -7,7 +7,7 @@
 from sglang.srt.utils import kill_child_process
 from sglang.test.test_utils import (
     DEFAULT_MODEL_NAME_FOR_TEST,
-    DEFAULT_URL_FOR_UNIT_TEST,
+    DEFAULT_URL_FOR_TEST,
     popen_launch_server,
 )
 
@@ -16,7 +16,7 @@ class TestJSONConstrained(unittest.TestCase):
     @classmethod
     def setUpClass(cls):
         cls.model = DEFAULT_MODEL_NAME_FOR_TEST
-        cls.base_url = DEFAULT_URL_FOR_UNIT_TEST
+        cls.base_url = DEFAULT_URL_FOR_TEST
         cls.api_key = "sk-123456"
         cls.json_schema = json.dumps(
             {