milvus/pkg/util/indexparamcheck/hnsw_checker.go
Jiquan Long 7be7e6f360
Refactor check logic of index parameters (#23856)
Signed-off-by: longjiquan <jiquan.long@zilliz.com>
2023-05-06 10:40:39 +08:00

22 lines
556 B
Go

package indexparamcheck
type hnswChecker struct {
floatVectorBaseChecker
}
func (c *hnswChecker) 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 newHnswChecker() IndexChecker {
return &hnswChecker{}
}