File tree Expand file tree Collapse file tree 1 file changed +17
-1
lines changed
Expand file tree Collapse file tree 1 file changed +17
-1
lines changed Original file line number Diff line number Diff 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
3147if __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" )
You can’t perform that action at this time.
0 commit comments