fix

Signed-off-by: BubbleCal <bubble-cal@outlook.com>

Author: BubbleCal

rust/lance-index/src/vector/ivf.rs

@@ -244,7 +244,10 @@ impl IvfTransformer {
 
     #[inline]
     pub fn compute_partitions(&self, data: &FixedSizeListArray) -> Result<UInt32Array> {
-        Ok(compute_partitions_arrow_array(&self.centroids, data, self.distance_type)?.into())
+        Ok(
+            compute_partitions_arrow_array(&self.centroids, data, self.distance_type)
+                .map(|(part_ids, _)| part_ids.into())?,
+        )
     }
 
     pub fn find_partitions(&self, query: &dyn Array, nprobes: usize) -> Result<UInt32Array> {

rust/lance-index/src/vector/residual.rs

@@ -77,6 +77,7 @@ where
             dimension,
             distance_type.expect("provide either partitions or distance type"),
         )
+        .0
         .into()
     });
     let part_ids = part_ids.values();