From 5668982dde56ef7b155c8c727529833804cf13ef Mon Sep 17 00:00:00 2001 From: Mark Hildebrand Date: Wed, 29 Jul 2026 08:30:38 -0700 Subject: [PATCH] Fix build failure. --- .github/workflows/ci.yml | 2 +- .github/workflows/nightly.yml | 2 +- diskann-inmem/integration/index/tests.rs | 2 +- diskann-inmem/src/provider.rs | 4 ++-- 4 files changed, 5 insertions(+), 5 deletions(-) diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index 5c78ce8eaa..a33b27a635 100644 --- a/.github/workflows/ci.yml +++ b/.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" diff --git a/.github/workflows/nightly.yml b/.github/workflows/nightly.yml index b5fb23be94..fab755b6af 100644 --- a/.github/workflows/nightly.yml +++ b/.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: diff --git a/diskann-inmem/integration/index/tests.rs b/diskann-inmem/integration/index/tests.rs index 436a851e3d..2ca4e73a07 100644 --- a/diskann-inmem/integration/index/tests.rs +++ b/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())?; diff --git a/diskann-inmem/src/provider.rs b/diskann-inmem/src/provider.rs index d7e01204e6..6167276186 100644 --- a/diskann-inmem/src/provider.rs +++ b/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;