Skip to content
2 changes: 1 addition & 1 deletion bigcode_eval/tasks/mbpp.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ def get_prompt(self, doc):
"""
description = doc["text"]
test_example = doc["test_list"][0]
prompt = f'"""\n{description}\n{test_example}\n"""\n'
prompt = f'"""\nText Description:\n\n{description}\n{test_example}\n"""\n\nCode:\n\n'
return prompt

def get_reference(self, doc):
Expand Down