Update instruction prompts with ntlk fix

Devy99 · Devy99 · commit 821f3a54e5b3 · 2025-04-10T21:49:30.000+02:00
diff --git a/tools/fix_v025.py b/tools/fix_v025.py
@@ -53,7 +53,7 @@ def map_ds(sample):
         sample['complete_prompt'] = "import nltk\nnltk.download('stopwords')\n" + sample['complete_prompt']
         sample['instruct_prompt'] = sample['instruct_prompt'].replace(
             "\nYou should write self-contained code starting with:\n```\n",
-            "\nYou should write self-contained code starting with:\n```\n"
+            "\nYou should write self-contained code starting with:\n```\nimport nltk\nnltk.download('stopwords')\n"
         )
 
     if sample["task_id"] in ["BigCodeBench/635"]:
@@ -68,16 +68,16 @@ def map_ds(sample):
         )
 
         sample['instruct_prompt'] = sample['instruct_prompt'].replace(
-            "# Importing the required libraries",
-            "# Importing the required libraries\nimport nltk\nnltk.download('stopwords')\n"
+            "\nYou should write self-contained code starting with:\n```\n",
+            "\nYou should write self-contained code starting with:\n```\nimport nltk\nnltk.download('stopwords')\n"
         )
 
     if sample["task_id"] in ["BigCodeBench/849"]:
         sample['code_prompt'] = "import nltk\nnltk.download('stopwords')\n" + sample['code_prompt']
         sample['complete_prompt'] = "import nltk\nnltk.download('stopwords')\n" + sample['complete_prompt']
         sample['instruct_prompt'] = sample['instruct_prompt'].replace(
             "\nYou should write self-contained code starting with:\n```\n",
-            "\nYou should write self-contained code starting with:\n```\n"
+            "\nYou should write self-contained code starting with:\n```\nimport nltk\nnltk.download('stopwords')\n"
         )
 
     if sample["task_id"] in ["BigCodeBench/940"]:

Original file line number	Diff line number	Diff line change
`@@ -53,7 +53,7 @@ def map_ds(sample):`
`53`	`53`	`sample['complete_prompt'] = "import nltk\nnltk.download('stopwords')\n" + sample['complete_prompt']`
`54`	`54`	`sample['instruct_prompt'] = sample['instruct_prompt'].replace(`
`55`	`55`	"\nYou should write self-contained code starting with:\n```\n",
`56`		- "\nYou should write self-contained code starting with:\n```\n"
	`56`	+ "\nYou should write self-contained code starting with:\n```\nimport nltk\nnltk.download('stopwords')\n"
`57`	`57`	`)`
`58`	`58`
`59`	`59`	`if sample["task_id"] in ["BigCodeBench/635"]:`
`@@ -68,16 +68,16 @@ def map_ds(sample):`
`68`	`68`	`)`
`69`	`69`
`70`	`70`	`sample['instruct_prompt'] = sample['instruct_prompt'].replace(`
`71`		`- "# Importing the required libraries",`
`72`		`- "# Importing the required libraries\nimport nltk\nnltk.download('stopwords')\n"`
	`71`	+ "\nYou should write self-contained code starting with:\n```\n",
	`72`	+ "\nYou should write self-contained code starting with:\n```\nimport nltk\nnltk.download('stopwords')\n"
`73`	`73`	`)`
`74`	`74`
`75`	`75`	`if sample["task_id"] in ["BigCodeBench/849"]:`
`76`	`76`	`sample['code_prompt'] = "import nltk\nnltk.download('stopwords')\n" + sample['code_prompt']`
`77`	`77`	`sample['complete_prompt'] = "import nltk\nnltk.download('stopwords')\n" + sample['complete_prompt']`
`78`	`78`	`sample['instruct_prompt'] = sample['instruct_prompt'].replace(`
`79`	`79`	"\nYou should write self-contained code starting with:\n```\n",
`80`		- "\nYou should write self-contained code starting with:\n```\n"
	`80`	+ "\nYou should write self-contained code starting with:\n```\nimport nltk\nnltk.download('stopwords')\n"
`81`	`81`	`)`
`82`	`82`
`83`	`83`	`if sample["task_id"] in ["BigCodeBench/940"]:`