This commit is contained in:
Dongsheng Zhu 2025-03-04 09:18:32 +00:00
parent a7070ba2d2
commit ff50937869

View File

@ -64,7 +64,7 @@ class LCBCodeGenerationDataset(BaseDataset):
# starter_code
if item['starter_code']:
format_prompt = f'### Format: {CodeGenerationPromptConstants.FORMATTING_MESSAGE_WITH_STARTER_CODE}\n' # noqa: E501
format_prompt += f"```python\n{item['starter_code']}\n```\n\n"
format_prompt += f"```python\n{item['starter_code']}\n```\n\n" # noqa: Q000, E501
else:
format_prompt = f'### Format: {CodeGenerationPromptConstants.FORMATTING_WITHOUT_STARTER_CODE}\n' # noqa: E501
format_prompt += '```python\n# YOUR CODE HERE\n```\n\n'