def formatting_prompts_func()

in use-cases/model-fine-tuning-pipeline/fine-tuning/pytorch/src/fine_tune.py [0:0]

• 6 lines of code
• 2 McCabe index (conditional complexity)

def formatting_prompts_func(example):
    output_texts = []
    for i in range(len(example["prompt"])):
        text = f"""{example["prompt"][i]}\n{EOS_TOKEN}"""
        output_texts.append(text)
    return {"prompts": output_texts}