[CI] Fix nightly test and raise better error message (#2626)
Co-authored-by: Sangbin <rkooo567@gmail.com>
This commit is contained in:
2
.github/workflows/nightly-test.yml
vendored
2
.github/workflows/nightly-test.yml
vendored
@@ -30,5 +30,5 @@ jobs:
|
|||||||
- name: Run test
|
- name: Run test
|
||||||
timeout-minutes: 10
|
timeout-minutes: 10
|
||||||
run: |
|
run: |
|
||||||
cd test/lang
|
cd test/srt
|
||||||
python3 run_suite.py --suite nightly --timeout-per-file 2400
|
python3 run_suite.py --suite nightly --timeout-per-file 2400
|
||||||
|
|||||||
@@ -484,7 +484,16 @@ def launch_engine(
|
|||||||
# Wait for model to finish loading
|
# Wait for model to finish loading
|
||||||
scheduler_infos = []
|
scheduler_infos = []
|
||||||
for i in range(len(scheduler_pipe_readers)):
|
for i in range(len(scheduler_pipe_readers)):
|
||||||
data = scheduler_pipe_readers[i].recv()
|
try:
|
||||||
|
data = scheduler_pipe_readers[i].recv()
|
||||||
|
except EOFError as e:
|
||||||
|
logger.exception(e)
|
||||||
|
logger.error(
|
||||||
|
f"Rank {i} scheduler is dead. Please check if there are relevant logs."
|
||||||
|
)
|
||||||
|
scheduler_procs[i].join()
|
||||||
|
logger.error(f"Exit code: {scheduler_procs[i].exitcode}")
|
||||||
|
raise
|
||||||
|
|
||||||
if data["status"] != "ready":
|
if data["status"] != "ready":
|
||||||
raise RuntimeError(
|
raise RuntimeError(
|
||||||
|
|||||||
@@ -44,11 +44,16 @@ suites = {
|
|||||||
"test_vision_openai_server.py",
|
"test_vision_openai_server.py",
|
||||||
"test_session_control.py",
|
"test_session_control.py",
|
||||||
],
|
],
|
||||||
|
"nightly": [
|
||||||
|
"test_nightly_gsm8k_eval.py",
|
||||||
|
"test_nightly_human_eval.py",
|
||||||
|
],
|
||||||
"sampling/penaltylib": glob.glob(
|
"sampling/penaltylib": glob.glob(
|
||||||
"sampling/penaltylib/**/test_*.py", recursive=True
|
"sampling/penaltylib/**/test_*.py", recursive=True
|
||||||
),
|
),
|
||||||
}
|
}
|
||||||
|
|
||||||
|
# Expand suite
|
||||||
for target_suite_name, target_tests in suites.items():
|
for target_suite_name, target_tests in suites.items():
|
||||||
for suite_name, tests in suites.items():
|
for suite_name, tests in suites.items():
|
||||||
if suite_name == target_suite_name:
|
if suite_name == target_suite_name:
|
||||||
|
|||||||
Reference in New Issue
Block a user