training

Documentation

Index

• Constants
• type MTFilter
• type Repository
• type ToolchainRepository

Constants

const (
	TagKey = "name"
)

Types

type MTFilter

type MTFilter struct {
	Toolchain    []string `name:"toolchain" postgres:"spec->>'toolchain'"`
	ModelName    []string `name:"model_name" postgres:"spec->'model'->>'name'"`
	ModelVersion []string `name:"model_version" postgres:"spec->'model'->>'version'"`
}

type Repository

type Repository interface {
	GetModelTraining(ctx context.Context, tx *sql.Tx, id string) (*training.ModelTraining, error)
	GetModelTrainingList(
		ctx context.Context, tx *sql.Tx, options ...filter.ListOption) ([]training.ModelTraining, error)
	DeleteModelTraining(ctx context.Context, tx *sql.Tx, id string) error
	SetDeletionMark(ctx context.Context, tx *sql.Tx, id string, value bool) error
	UpdateModelTraining(ctx context.Context, tx *sql.Tx, mt *training.ModelTraining) error
	UpdateModelTrainingStatus(ctx context.Context, tx *sql.Tx, id string, s v1alpha1.ModelTrainingStatus) error
	SaveModelTraining(ctx context.Context, tx *sql.Tx, mt *training.ModelTraining) error
	BeginTransaction(ctx context.Context) (*sql.Tx, error)
}

type ToolchainRepository

type ToolchainRepository interface {
	GetToolchainIntegration(name string) (*training.ToolchainIntegration, error)
	GetToolchainIntegrationList(options ...filter.ListOption) ([]training.ToolchainIntegration, error)
	DeleteToolchainIntegration(name string) error
	UpdateToolchainIntegration(md *training.ToolchainIntegration) error
	SaveToolchainIntegration(md *training.ToolchainIntegration) error
}