[CI] Fix CI (#1217)

This commit is contained in:
Mingyi
2024-08-25 19:56:42 -07:00
committed by GitHub
parent 158e8f1e2d
commit 7514b9f8d3
5 changed files with 22 additions and 27 deletions

View File

@@ -77,4 +77,4 @@ if __name__ == "__main__":
except RuntimeError:
pass
unittest.main(warnings="ignore")
unittest.main()

View File

@@ -136,4 +136,4 @@ if __name__ == "__main__":
except RuntimeError:
pass
unittest.main(warnings="ignore")
unittest.main()

View File

@@ -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)

View File

@@ -112,4 +112,4 @@ class TestBatchPenalizerE2E(unittest.TestCase):
if __name__ == "__main__":
unittest.main(warnings="ignore")
unittest.main()