diff --git a/python/sglang/srt/managers/schedule_policy.py b/python/sglang/srt/managers/schedule_policy.py index 3cd2be26b..1dd07881e 100644 --- a/python/sglang/srt/managers/schedule_policy.py +++ b/python/sglang/srt/managers/schedule_policy.py @@ -22,11 +22,7 @@ from typing import Dict, List, Optional, Set, Union import torch -from sglang.srt.managers.schedule_batch import ( - Req, - ScheduleBatch, - global_server_args_dict, -) +from sglang.srt.managers.schedule_batch import Req, ScheduleBatch from sglang.srt.mem_cache.base_prefix_cache import BasePrefixCache from sglang.srt.mem_cache.memory_pool import TokenToKVPoolAllocator from sglang.srt.mem_cache.radix_cache import RadixCache, TreeNode diff --git a/test/srt/test_pp_single_node.py b/test/srt/test_pp_single_node.py index 4588c1326..8a4ffaa62 100644 --- a/test/srt/test_pp_single_node.py +++ b/test/srt/test_pp_single_node.py @@ -4,7 +4,6 @@ python3 -m unittest test_pp_single_node.TestPPAccuracy.test_gsm8k python3 -m unittest test_pp_single_node.TestFixedBugs.test_chunked_prefill_with_small_bs """ -import os import time import unittest from types import SimpleNamespace