INNER CODE UNIT · Go
getRepositoryHandler
SeldonIO/seldon-core · scheduler/cmd/agent/main.go:66
func getRepositoryHandler(logger log.FieldLogger) repository.ModelRepositoryHandler {
switch cli.ServerType {
case "mlserver":
logger.Infof("Creating MLServer repository handler")
return mlserver.NewMLServerRepositoryHandler(logger)
case "triton":
logger.Infof("Creating Triton repository handler")
return triton.NewTritonRepositoryHandler(logger)
default:
logger.Infof("Using default as no server type requested - creating MLServer repository handler")
return mlserver.NewMLServerRepositoryHandler(logger)
}
}
func createReplicaConfig() *agent2.ReplicaConfig {
var rc *agent2.ReplicaConfig
if cli.ReplicaConfigStr != "" {
var err error