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}