()
| 152 | } |
| 153 | |
| 154 | fn trained_codec() -> Sq8Rerank { |
| 155 | let samples: Vec<Vec<f32>> = (0..50).map(|i| make_vec(i as f32 * 0.1)).collect(); |
| 156 | let refs: Vec<&[f32]> = samples.iter().map(|v| v.as_slice()).collect(); |
| 157 | let mut codec = Sq8Rerank::new(DIM); |
| 158 | codec.train(&refs).expect("train must succeed"); |
| 159 | codec |
| 160 | } |
| 161 | |
| 162 | #[test] |
| 163 | fn round_trip_returns_finite_distance() { |