Skip to content

Commit 1e12249

Browse files
committed
fix gen file name
1 parent 0f30a64 commit 1e12249

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

bigcodebench/generate.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -188,7 +188,11 @@ def run_codegen(
188188
extra = "-" + subset if subset != "full" else ""
189189
if reasoning_effort and model.startswith("o1-") or model.startswith("o3-") or model.endswith("-reasoner"):
190190
model = model + f"--{reasoning_effort}"
191-
identifier = model.replace("/", "--") + "--skip_prefill" if skip_prefill else "" + f"--{revision}--bigcodebench{extra}-{split}--{backend}-{temperature}-{n_samples}-sanitized_calibrated.jsonl"
191+
192+
if skip_prefill:
193+
identifier = model.replace("/", "--") + "--skip_prefill" + f"--{revision}--bigcodebench{extra}-{split}--{backend}-{temperature}-{n_samples}-sanitized_calibrated.jsonl"
194+
else:
195+
identifier = model.replace("/", "--") + f"--{revision}--bigcodebench{extra}-{split}--{backend}-{temperature}-{n_samples}-sanitized_calibrated.jsonl"
192196

193197
target_path = os.path.join(root, identifier)
194198

0 commit comments

Comments
 (0)