milvus/pkg/util/indexparamcheck/diskann_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

18 lines
460 B
Go

package indexparamcheck
// diskannChecker checks if an diskann index can be built.
type diskannChecker struct {
floatVectorBaseChecker
}
func (c *diskannChecker) CheckTrain(params map[string]string) error {
if !CheckIntByRange(params, DIM, DiskAnnMinDim, DefaultMaxDim) {
return errOutOfRange(DIM, DiskAnnMinDim, DefaultMaxDim)
}
return c.floatVectorBaseChecker.CheckTrain(params)
}
func newDiskannChecker() IndexChecker {
return &diskannChecker{}
}