def prepare_data()

in cookbook-efforts/kto-preference/preference_gen.py [0:0]

• 4 lines of code
• 1 McCabe index (conditional complexity)

def prepare_data():
    prompts = load_dataset("davanstrien/haiku_prompts", split="train")
    print(f"Loaded {len(prompts)} prompts")
    return prompts.rename_column("instructions", "input")