Skip to content

Commit e71fe2e

Browse files
committed
fix (data): update 241 & 267
1 parent b0676a7 commit e71fe2e

File tree

1 file changed

+17
-1
lines changed

1 file changed

+17
-1
lines changed

tools/fix_v0110.py

Lines changed: 17 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,22 @@ def map_ds(sample):
2626
"Requirements:\n - pandas\n - sklearn.ensemble\n"
2727
)
2828

29+
if sample["task_id"] in ["BigCodeBench/241"]:
30+
for k in sample.keys():
31+
if "prompt" in k:
32+
sample[k] = sample[k].replace(
33+
"The function will plot the original and normalized arrays using matplotlib.",
34+
"The function will plot the original and normalized arrays with a title of 'Original vs. Normalized Data'."
35+
)
36+
37+
if sample["task_id"] in ["BigCodeBench/267"]:
38+
for k in sample.keys():
39+
if "prompt" in k:
40+
sample[k] = sample[k].replace(
41+
"Plots and returns the FFT of the signal.",
42+
"Plots and returns the FFT of the signal with a title of 'FFT of the signal'."
43+
)
44+
2945
return sample
3046

3147
if __name__ == "__main__":
@@ -34,7 +50,7 @@ def map_ds(sample):
3450
hard_ds_dict = load_dataset(BIGCODEBENCH_HARD_HF)
3551
ds = ds_dict[BIGCODEBENCH_VERSION]
3652
hard_ds = hard_ds_dict[BIGCODEBENCH_VERSION]
37-
function_id = [16, 37]
53+
function_id = [16, 37, 241, 267]
3854

3955
new_ds = ds.map(map_ds)
4056
new_ds.to_json("BigCodeBench.jsonl")

0 commit comments

Comments
 (0)