Fix Build

.github/workflows/ci.yml

@@ -22,7 +22,7 @@ env:
   CARGO_TERM_COLOR: always
   # The features we want to explicitly test. For example, the `flatbuffers-build` feature
   # of `diskann-quantization` requires additional setup and so must not be included by default.
-  DISKANN_FEATURES: "virtual_storage,spherical-quantization,product-quantization,tracing,experimental_diversity_search,disk-index,flatbuffers,linalg,codegen,multi-vector,bftree,inmem2"
+  DISKANN_FEATURES: "virtual_storage,spherical-quantization,product-quantization,tracing,experimental_diversity_search,disk-index,flatbuffers,linalg,codegen,multi-vector,bftree,inmem2,integration-test"
 
   # Intel SDE version used for baseline and AVX-512 emulation jobs.
   SDE_VERSION: "sde-external-10.8.0-2026-03-15-lin"

.github/workflows/nightly.yml

@@ -17,7 +17,7 @@ env:
   RUST_CONFIG: 'build.rustflags=["-Dwarnings"]'
   RUST_BACKTRACE: 1
   CARGO_TERM_COLOR: always
-  DISKANN_FEATURES: "virtual_storage,spherical-quantization,product-quantization,tracing,experimental_diversity_search,disk-index,flatbuffers,linalg,codegen,multi-vector,bftree,inmem2"
+  DISKANN_FEATURES: "virtual_storage,spherical-quantization,product-quantization,tracing,experimental_diversity_search,disk-index,flatbuffers,linalg,codegen,multi-vector,bftree,inmem2,integration-test"
 
 defaults:
   run:

diskann-inmem/integration/index/tests.rs

@@ -54,7 +54,7 @@ pub(super) fn knn(
         let stats = rt.block_on(index.search(query, knn, &mut neighbors))?;
         misc += stats;
 
-        std::iter::zip(out.iter_mut(), neighbors.iter()).for_each(|(d, s)| *d = s.id);
+        std::iter::zip(out.iter_mut(), neighbors.iter()).for_each(|(d, s)| *d = *s.id());
     }
     let counters = before.delta(&index.counters())?;
 

diskann-inmem/src/provider.rs

@@ -939,8 +939,8 @@ where
 
             let mut count = 0;
             for c in candidates {
-                if let Some(ext) = provider.mapping.to_external(c.id) {
-                    if output.push(ext, c.distance).is_available() {
+                if let Some(ext) = provider.mapping.to_external(*c.id()) {
+                    if output.push(ext, c.distance()).is_available() {
                         count += 1;
                     } else {
                         break;