diff --git a/python/sglang/srt/managers/tp_worker.py b/python/sglang/srt/managers/tp_worker.py index 896f4226db8..127f71900ae 100644 --- 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 index 5eda6bc3a3d..5393ecc33ca 100644 --- 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( {