adapt to sglang v0.5.2rc1 on dcu
This commit is contained in:
20
sgl-router/py_test/run_suite.py
Normal file
20
sgl-router/py_test/run_suite.py
Normal file
@@ -0,0 +1,20 @@
|
||||
import argparse
|
||||
import glob
|
||||
|
||||
from sglang.test.test_utils import TestFile, run_unittest_files
|
||||
|
||||
if __name__ == "__main__":
|
||||
arg_parser = argparse.ArgumentParser()
|
||||
arg_parser.add_argument(
|
||||
"--timeout-per-file",
|
||||
type=int,
|
||||
default=2000,
|
||||
help="The time limit for running one file in seconds.",
|
||||
)
|
||||
args = arg_parser.parse_args()
|
||||
|
||||
files = glob.glob("**/test_*.py", recursive=True)
|
||||
|
||||
test_files = [TestFile(name=file) for file in files]
|
||||
exit_code = run_unittest_files(test_files, args.timeout_per_file)
|
||||
exit(exit_code)
|
||||
Reference in New Issue
Block a user