diff --git a/core/src/wrapper/ConfAdapter.cpp b/core/src/wrapper/ConfAdapter.cpp index a94fdac8b2..d706048cf2 100644 --- a/core/src/wrapper/ConfAdapter.cpp +++ b/core/src/wrapper/ConfAdapter.cpp @@ -18,6 +18,7 @@ #include "wrapper/ConfAdapter.h" #include "knowhere/index/vector_index/helpers/IndexParameter.h" #include "utils/Log.h" +#include "WrapperException.h" #include #include @@ -146,8 +147,10 @@ IVFPQConfAdapter::MatchSearch(const TempMetaConf& metaconf, const IndexType& typ auto conf = std::make_shared(); conf->k = metaconf.k; - if (metaconf.nprobe <= 0) - conf->nprobe = 16; // hardcode here + if (metaconf.nprobe <= 0){ + WRAPPER_LOG_ERROR << "The nprobe of PQ is wrong!"; + throw WrapperException("The nprobe of PQ is wrong!"); + } else conf->nprobe = metaconf.nprobe; diff --git a/core/unittest/wrapper/test_wrapper.cpp b/core/unittest/wrapper/test_wrapper.cpp index c12bd5ea42..025601a1cd 100644 --- a/core/unittest/wrapper/test_wrapper.cpp +++ b/core/unittest/wrapper/test_wrapper.cpp @@ -51,6 +51,7 @@ class KnowhereWrapperTest tempconf.size = nb; tempconf.dim = dim; tempconf.k = k; + tempconf.nprobe = 16; index_ = GetVecIndexFactory(index_type); conf = ParamGenerator::GetInstance().GenBuild(index_type, tempconf);