[CI] Fix CI (#1217)
This commit is contained in:
@@ -77,4 +77,4 @@ if __name__ == "__main__":
|
||||
except RuntimeError:
|
||||
pass
|
||||
|
||||
unittest.main(warnings="ignore")
|
||||
unittest.main()
|
||||
|
||||
@@ -136,4 +136,4 @@ if __name__ == "__main__":
|
||||
except RuntimeError:
|
||||
pass
|
||||
|
||||
unittest.main(warnings="ignore")
|
||||
unittest.main()
|
||||
|
||||
@@ -7,7 +7,7 @@ from sglang.test.test_utils import run_unittest_files
|
||||
suites = {
|
||||
"minimal": [
|
||||
"models/test_embedding_models.py",
|
||||
# "models/test_generation_models.py",
|
||||
"models/test_generation_models.py",
|
||||
"sampling/penaltylib",
|
||||
"test_chunked_prefill.py",
|
||||
"test_embedding_openai_server.py",
|
||||
@@ -33,6 +33,7 @@ for target_suite_name, target_tests in suites.items():
|
||||
tests.remove(target_suite_name)
|
||||
tests.extend(target_tests)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
arg_parser = argparse.ArgumentParser()
|
||||
arg_parser.add_argument(
|
||||
@@ -55,10 +56,5 @@ if __name__ == "__main__":
|
||||
else:
|
||||
files = suites[args.suite]
|
||||
|
||||
try:
|
||||
mp.set_start_method("spawn")
|
||||
except RuntimeError:
|
||||
pass
|
||||
|
||||
exit_code = run_unittest_files(files, args.timeout_per_file)
|
||||
exit(exit_code)
|
||||
|
||||
@@ -112,4 +112,4 @@ class TestBatchPenalizerE2E(unittest.TestCase):
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main(warnings="ignore")
|
||||
unittest.main()
|
||||
|
||||
Reference in New Issue
Block a user