repository

Documentation

Index

• type ModelRepository
• type ModelRepositoryHandler
• type V2ModelRepository
  • func NewModelRepository(logger log.FieldLogger, rcloneClient *rclone.RCloneClient, repoPath string, ...) *V2ModelRepository
  • func (r *V2ModelRepository) DownloadModelVersion(modelName string, version uint32, artifactVersion *uint32, srcUri string, ...) (*string, error)
  • func (r *V2ModelRepository) Ready() error
  • func (r *V2ModelRepository) RemoveModelVersion(modelName string) error

Types

type ModelRepository

type ModelRepository interface {
	DownloadModelVersion(modelName string, version uint32, artifactVersion *uint32, srcUri string, config []byte, explainerSpec *scheduler.ExplainerSpec, parameters []*scheduler.ParameterSpec) (*string, error)
	RemoveModelVersion(modelName string) error
	Ready() error
}

type ModelRepositoryHandler

type ModelRepositoryHandler interface {
	FindModelVersionFolder(modelName string, version *uint32, path string) (string, error)
	UpdateModelVersion(modelName string, version uint32, path string) error
	UpdateModelRepository(modelName string, versionPath, modelRepoPath string) error
	SetExplainer(modelRepoPath string, explainerSpec *scheduler.ExplainerSpec, envoyHost string, envoyPort int) error
	SeExtratParameters(modelRepoPath string, parameters []*scheduler.ParameterSpec) error
}

type V2ModelRepository

type V2ModelRepository struct {
	// contains filtered or unexported fields
}

func NewModelRepository

func NewModelRepository(logger log.FieldLogger,
	rcloneClient *rclone.RCloneClient,
	repoPath string,
	modelRepositoryHandler ModelRepositoryHandler,
	envoyHost string,
	envoyPort int) *V2ModelRepository

func (*V2ModelRepository) DownloadModelVersion

func (r *V2ModelRepository) DownloadModelVersion(modelName string,
	version uint32,
	artifactVersion *uint32,
	srcUri string,
	config []byte,
	explainerSpec *scheduler.ExplainerSpec,
	parameters []*scheduler.ParameterSpec) (*string, error)

func (*V2ModelRepository) Ready

func (r *V2ModelRepository) Ready() error

func (*V2ModelRepository) RemoveModelVersion

func (r *V2ModelRepository) RemoveModelVersion(modelName string) error

Remove version folder and return number of remaining versions calculated as found model-settings files