store

Documentation

Index

• Constants
• type RedisStore
  • func (r *RedisStore) AddModel(ctx context.Context, name string, m types.ModelEntry, tags ...string) error
  • func (r *RedisStore) Artifact(ctx context.Context, runID string, id string) (types.SavedArtifact, error)
  • func (r *RedisStore) ArtifactExist(ctx context.Context, runID string, id string) error
  • func (r *RedisStore) Artifacts(ctx context.Context, runID string) (map[string]types.SavedArtifact, error)
  • func (r *RedisStore) ArtifactsExist(ctx context.Context, runID string, ids []string) map[string]error
  • func (r *RedisStore) CreateModelRegistry(ctx context.Context, m types.ModelRegistry) error
  • func (r *RedisStore) Exp(ctx context.Context, expID string) (*types.Experiment, error)
  • func (r *RedisStore) ExpExists(ctx context.Context, expID string) error
  • func (r *RedisStore) ExpRunIDs(ctx context.Context, expID string) ([]string, error)
  • func (r *RedisStore) Exps(ctx context.Context) ([]string, error)
  • func (r *RedisStore) LastModel(ctx context.Context, name string) (types.ModelEntry, error)
  • func (r *RedisStore) Metrics(ctx context.Context, runID string) ([]types.Metric, error)
  • func (r *RedisStore) ModelByTag(ctx context.Context, name string, tag string) (types.ModelEntry, error)
  • func (r *RedisStore) ModelByVersion(ctx context.Context, name string, version int) (types.ModelEntry, error)
  • func (r *RedisStore) ModelRegistry(ctx context.Context, name string) (*types.ModelRegistry, error)
  • func (r *RedisStore) ModelRegistryExists(ctx context.Context, name string) error
  • func (r *RedisStore) Run(ctx context.Context, id string) (*types.Run, error)
  • func (r *RedisStore) RunExists(ctx context.Context, runID string) (bool, error)
  • func (r *RedisStore) RunMetrics(ctx context.Context, runID string) ([]string, error)
  • func (r *RedisStore) Runs(ctx context.Context, ids []string) ([]*types.Run, error)
  • func (r *RedisStore) SetArtifact(ctx context.Context, runID string, a types.SavedArtifact) error
  • func (r *RedisStore) SetArtifacts(ctx context.Context, runID string, as []types.SavedArtifact) error
  • func (r *RedisStore) SetMetric(ctx context.Context, runID string, m types.Metric) error
  • func (r *RedisStore) SetMetrics(ctx context.Context, runID string, ms map[string]types.Metric) error
  • func (r *RedisStore) SetRun(ctx context.Context, run types.Run) error
  • func (r *RedisStore) UpdateModelRegistry(ctx context.Context, m types.ModelRegistry) error

Constants

const (
	ExpKeyPattern = "exp:%s"
	// RunKeyPattern pattern of a Run's key
	// Example:
	// run:linear-regression
	RunKeyPattern = "run:%s"
	// MetricKeyPattern pattern of a metric metric:<metric_name>:<run_id>
	// Example
	// metric:mse:linear-regression
	MetricKeyPattern = "metric:%s:%s"
	// ArtifactKeyPattern pattern of a artifact's key
	// Example
	// artifact:logs:linear-regression
	ArtifactKeyPattern = "artifact:%s:%s"

	ModelRegistryInfoKeyPattern = "info:registry:%s"
	// ModelRegistryKeyPattern pattern a model registry key
	// Example
	// registry:yolov12
	ModelRegistryKeyPattern = "registry:%s"
	// ModelRegistryTagsKeyPattern pattern a model registry's tags
	// Example
	// tag:registry:yolov12 -> [prod, latest, ...]
	ModelRegistryTagsKeyPattern = "tag:registry:%s"
	// ModelRegistryTagKeyPattern pattern a model registry tag
	// Example
	// tag:registry:yolov12:prod
	ModelRegistryTagKeyPattern = "tag:registry:%s:%s"

	TransactionMaxTries = 10
)

Types

type RedisStore

type RedisStore struct {
	Client redis.Client
}

func (*RedisStore) AddModel

func (r *RedisStore) AddModel(ctx context.Context, name string, m types.ModelEntry, tags ...string) error

func (*RedisStore) Artifact

func (r *RedisStore) Artifact(ctx context.Context, runID string, id string) (types.SavedArtifact, error)

func (*RedisStore) ArtifactExist

func (r *RedisStore) ArtifactExist(ctx context.Context, runID string, id string) error

func (*RedisStore) Artifacts

func (r *RedisStore) Artifacts(ctx context.Context, runID string) (map[string]types.SavedArtifact, error)

func (*RedisStore) ArtifactsExist

func (r *RedisStore) ArtifactsExist(ctx context.Context, runID string, ids []string) map[string]error

func (*RedisStore) CreateModelRegistry

func (r *RedisStore) CreateModelRegistry(ctx context.Context, m types.ModelRegistry) error

func (*RedisStore) Exp

func (r *RedisStore) Exp(ctx context.Context, expID string) (*types.Experiment, error)

func (*RedisStore) ExpExists

func (r *RedisStore) ExpExists(ctx context.Context, expID string) error

func (*RedisStore) ExpRunIDs

func (r *RedisStore) ExpRunIDs(ctx context.Context, expID string) ([]string, error)

func (*RedisStore) Exps

func (r *RedisStore) Exps(ctx context.Context) ([]string, error)

func (*RedisStore) LastModel

func (r *RedisStore) LastModel(ctx context.Context, name string) (types.ModelEntry, error)

func (*RedisStore) Metrics

func (r *RedisStore) Metrics(ctx context.Context, runID string) ([]types.Metric, error)

func (*RedisStore) ModelByTag

func (r *RedisStore) ModelByTag(ctx context.Context, name string, tag string) (types.ModelEntry, error)

func (*RedisStore) ModelByVersion

func (r *RedisStore) ModelByVersion(ctx context.Context, name string, version int) (types.ModelEntry, error)

func (*RedisStore) ModelRegistry

func (r *RedisStore) ModelRegistry(ctx context.Context, name string) (*types.ModelRegistry, error)

func (*RedisStore) ModelRegistryExists

func (r *RedisStore) ModelRegistryExists(ctx context.Context, name string) error

ModelRegistryExists returns non-nil error if registry was not found.

func (*RedisStore) Run

func (r *RedisStore) Run(ctx context.Context, id string) (*types.Run, error)

func (*RedisStore) RunExists

func (r *RedisStore) RunExists(ctx context.Context, runID string) (bool, error)

func (*RedisStore) RunMetrics

func (r *RedisStore) RunMetrics(ctx context.Context, runID string) ([]string, error)

func (*RedisStore) Runs

func (r *RedisStore) Runs(ctx context.Context, ids []string) ([]*types.Run, error)

func (*RedisStore) SetArtifact

func (r *RedisStore) SetArtifact(ctx context.Context, runID string, a types.SavedArtifact) error

func (*RedisStore) SetArtifacts

func (r *RedisStore) SetArtifacts(ctx context.Context, runID string, as []types.SavedArtifact) error

func (*RedisStore) SetMetric

func (r *RedisStore) SetMetric(ctx context.Context, runID string, m types.Metric) error

func (*RedisStore) SetMetrics

func (r *RedisStore) SetMetrics(ctx context.Context, runID string, ms map[string]types.Metric) error

func (*RedisStore) SetRun

func (r *RedisStore) SetRun(ctx context.Context, run types.Run) error

func (*RedisStore) UpdateModelRegistry

func (r *RedisStore) UpdateModelRegistry(ctx context.Context, m types.ModelRegistry) error