distill.rs

 1use anyhow::Result;
 2use std::mem;
 3
 4use crate::example::Example;
 5
 6pub async fn run_distill(example: &mut Example) -> Result<()> {
 7    let predictions = mem::take(&mut example.predictions)
 8        .into_iter()
 9        .map(|p| p.actual_patch)
10        .collect();
11
12    example.spec.expected_patches = predictions;
13    example.prompt = None;
14    example.predictions = Vec::new();
15    example.score = Vec::new();
16    Ok(())
17}