Skip to content

Commit 524d239

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

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

tests/e2e/singlecard/models/test_lm_eval_correctness.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -90,6 +90,14 @@ 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+
94+
dir_path = os.path.dirname(report_output)
95+
if os.path.exists(dir_path):
96+
print(f"✅ 目录已确保存在:{dir_path}")
97+
else:
98+
print(f"❌ 目录创建失败:{dir_path}")
99+
100+
os.makedirs(os.path.dirname(report_output), exist_ok=True)
93101
with open(report_output, 'w', encoding='utf-8') as f:
94102
f.write(report_content)
95103

0 commit comments

Comments
 (0)