Fix RHNSWPQ pqm divide by zero ()

See also: 

Signed-off-by: yangxuan <xuan.yang@zilliz.com>

Signed-off-by: yangxuan <xuan.yang@zilliz.com>
pull/18725/head
XuanYang-cn 2022-08-18 19:24:51 +08:00 committed by GitHub
parent cf1c5902e5
commit a782ded0bd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 1 deletions
internal/util/indexparamcheck

View File

@ -419,7 +419,7 @@ func (adapter *RHNSWPQConfAdapter) CheckTrain(params map[string]string) bool {
return false
}
pqm, err := strconv.Atoi(pqmStr)
if err != nil {
if err != nil || pqm == 0 {
return false
}

View File

@ -461,6 +461,9 @@ func TestRHNSWPQConfAdapter_CheckTrain(t *testing.T) {
invalidParamsPQM := copyParams(validParams)
invalidParamsPQM[PQM] = "NAN"
invalidParamsPQMZero := copyParams(validParams)
invalidParamsPQMZero[PQM] = "0"
cases := []struct {
params map[string]string
want bool
@ -473,6 +476,7 @@ func TestRHNSWPQConfAdapter_CheckTrain(t *testing.T) {
{invalidMParamsMax, false},
{invalidParamsWithoutPQM, false},
{invalidParamsPQM, false},
{invalidParamsPQMZero, false},
}
adapter := newRHNSWPQConfAdapter()