diff --git a/cpp/test/neighbors/ann_cagra.cuh b/cpp/test/neighbors/ann_cagra.cuh index 1d1f244db3..499abd7e26 100644 --- a/cpp/test/neighbors/ann_cagra.cuh +++ b/cpp/test/neighbors/ann_cagra.cuh @@ -15,7 +15,6 @@ */ #pragma once -#include "raft/neighbors/ivf_pq_types.hpp" #undef RAFT_EXPLICIT_INSTANTIATE_ONLY // Search with filter instantiation #include "../test_utils.cuh" @@ -29,6 +28,7 @@ #include #include #include +#include #include #include #include @@ -246,7 +246,8 @@ class AnnCagraTest : public ::testing::TestWithParam { protected: void testCagra() { - // TODO (tarang-jain): remove when NN Descent index building support InnerProduct + // TODO (tarang-jain): remove when NN Descent index building support InnerProduct. Reference + // issue: https://github.com/rapidsai/raft/issues/2276 if (ps.metric == distance::InnerProduct && ps.build_algo == graph_build_algo::NN_DESCENT) GTEST_SKIP();