[CI] balance unit tests (#1977)

This commit is contained in:
Lianmin Zheng
2024-11-09 16:46:14 -08:00
committed by GitHub
parent 9c939a3d8b
commit 520f0094e4
4 changed files with 25 additions and 16 deletions

View File

@@ -442,7 +442,7 @@ def popen_launch_server(
"Content-Type": "application/json; charset=utf-8",
"Authorization": f"Bearer {api_key}",
}
response = requests.get(f"{base_url}/v1/models", headers=headers)
response = requests.get(f"{base_url}/health_generate", headers=headers)
if response.status_code == 200:
return process
except requests.RequestException:
@@ -637,8 +637,8 @@ def calculate_rouge_l(output_strs_list1, output_strs_list2):
return rouge_l_scores
STDOUT_FILENAME = "stdout.txt"
STDERR_FILENAME = "stderr.txt"
STDOUT_FILENAME = "stdout.txt"
def read_output(output_lines):