From 1c8caee7a8193df2b5341551c7c6608eccf94960 Mon Sep 17 00:00:00 2001
From: mannaandpoem <1580466765@qq.com>
Date: Mon, 18 Mar 2024 14:58:23 +0800
Subject: [PATCH] Add Inference of SWE-bench for Data Interpreter

---
 data/inference/run_api.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/data/inference/run_api.py b/data/inference/run_api.py
index 4fa2f5894..18fb48c40 100644
--- a/data/inference/run_api.py
+++ b/data/inference/run_api.py
@@ -106,6 +106,7 @@ async def main(
         model_nickname = Path(model_name_or_path).name
     output_file = f"{model_nickname}__{dataset_name_or_path.split('/')[-1]}__{split}"
     output_file = Path(output_dir, output_file + ".jsonl")
+    output_file.parent.mkdir(parents=True, exist_ok=True)
     logger.info(f"Will write to {output_file}")
     existing_ids = set()
     if os.path.exists(output_file):