From dbfa98b08436ffdca71ce992024fec7178fc2d81 Mon Sep 17 00:00:00 2001 From: BubbleCal Date: Sat, 8 Feb 2025 15:31:36 +0800 Subject: [PATCH] fix Signed-off-by: BubbleCal --- rust/lance/src/io/exec/knn.rs | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/rust/lance/src/io/exec/knn.rs b/rust/lance/src/io/exec/knn.rs index d26ce6f14c..017df7bbc1 100644 --- a/rust/lance/src/io/exec/knn.rs +++ b/rust/lance/src/io/exec/knn.rs @@ -715,8 +715,7 @@ impl ExecutionPlan for MultivectorScoringExec { self: Arc, children: Vec>, ) -> DataFusionResult> { - let plan = - MultivectorScoringExec::try_new(self.dataset.clone(), children, self.query.clone())?; + let plan = Self::try_new(self.dataset.clone(), children, self.query.clone())?; Ok(Arc::new(plan)) } @@ -792,9 +791,7 @@ impl ExecutionPlan for MultivectorScoringExec { } }); query_results.into_iter().for_each(|(row_id, sim)| { - if !results.contains_key(&row_id) { - results.insert(row_id, sim + missed_similarities); - } + results.entry(row_id).or_insert(sim + missed_similarities); }); missed_similarities += min_sim; }