Skip to content

Commit 8948ac4

Browse files
committed
fix
Signed-off-by: Icey <1790571317@qq.com>
1 parent cbb1ed1 commit 8948ac4

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/e2e/singlecard/models/test_lm_eval_correctness.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ def build_eval_args(eval_config, tp_size):
6868

6969

7070
def generate_report(tp_size, eval_config, report_data, report_template,
71-
output_path, env_config):
71+
report_output, env_config):
7272
env = Environment(loader=FileSystemLoader('.'))
7373
template = env.get_template(str(report_template))
7474
model_args = build_model_args(eval_config, tp_size)
@@ -90,12 +90,12 @@ def generate_report(tp_size, eval_config, report_data, report_template,
9090
batch_size="auto",
9191
num_fewshot=eval_params["num_fewshot"],
9292
rows=report_data["rows"])
93-
with open(output_path, 'w', encoding='utf-8') as f:
93+
with open(report_output, 'w', encoding='utf-8') as f:
9494
f.write(report_content)
9595

9696

9797
def test_lm_eval_correctness_param(config_filename, tp_size, report_template,
98-
output_path, env_config):
98+
report_output, env_config):
9999
eval_config = yaml.safe_load(config_filename.read_text(encoding="utf-8"))
100100
eval_params = build_eval_args(eval_config, tp_size)
101101
results = lm_eval.simple_evaluate(**eval_params)
@@ -123,5 +123,5 @@ def test_lm_eval_correctness_param(config_filename, tp_size, report_template,
123123
',', '_stderr,', 1)]
124124
})
125125
generate_report(tp_size, eval_config, report_data, report_template,
126-
output_path, env_config)
126+
report_output, env_config)
127127
assert success

0 commit comments

Comments
 (0)