milvus/internal/util/indexparamcheck/rhnsw_checker.go
Jiquan Long a8306599db
Refactor check logic of index parameters (#23856) (#23903)
Signed-off-by: longjiquan <jiquan.long@zilliz.com>
2023-05-09 14:20:42 +08:00

22 lines
575 B
Go

package indexparamcheck
type rHnswFlatChecker struct {
floatVectorBaseChecker
}
func (c rHnswFlatChecker) CheckTrain(params map[string]string) error {
if !CheckIntByRange(params, EFConstruction, HNSWMinEfConstruction, HNSWMaxEfConstruction) {
return errOutOfRange(EFConstruction, HNSWMinEfConstruction, HNSWMaxEfConstruction)
}
if !CheckIntByRange(params, HNSWM, HNSWMinM, HNSWMaxM) {
return errOutOfRange(HNSWM, HNSWMinM, HNSWMaxM)
}
return c.floatVectorBaseChecker.CheckTrain(params)
}
func newRHnswFlatChecker() IndexChecker {
return &rHnswFlatChecker{}
}