distill.rs

 1use std::mem;
 2
 3use crate::example::Example;
 4
 5pub async fn run_distill(example: &mut Example) {
 6    let [prediction]: [_; 1] = mem::take(&mut example.predictions)
 7        .try_into()
 8        .expect("Run predict first with a single repetition");
 9
10    example.expected_patch = prediction.actual_patch;
11    example.prompt = None;
12    example.predictions = Vec::new();
13    example.score = Vec::new();
14}