mlflow

type ArtifactRepo interface {
	// localPath is the path to the file on the local filesystem.
	// artifactPath is the directory in the artifact repo to upload the file to.
	// Kinda weird, but this is how the python client does it.
	LogArtifact(localPath, artifactPath string) error
	// LogArtifacts logs all of the files in a directory tree as artifacts.
	// localPath is the path to the directory on the local filesystem.
	// artifactPath is the directory in the artifact repo to upload to.
	LogArtifacts(localDir, artifactPath string) error
}

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

type Experiment interface {
	CreateRun(name string) (Run, error)
	GetRun(runId string) (Run, error)
	ID() string
}

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

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

type Metric struct {
	Key string
	Val float64
}

type Param struct {
	Key string
	Val string
}

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

type Run interface {
	SetName(name string) error
	Name() string
	SetTag(key, value string) error
	SetTags(tags []Tag) error
	GetTag(key string) (string, error)
	LogArtifact(localPath, artifactPath string) error
	LogMetric(key string, val float64, step int64) error
	LogMetrics(metrics []Metric, step int64) error
	LogParam(key, value string) error
	LogParams(params []Param) error
	GetParam(key string) (string, error)
	End() error
	Fail() error
	UIURL() string
	ID() string
	ExperimentID() string
}

type Tag struct {
	Key string
	Val string
}

type Tracking interface {
	ExperimentsByName() (map[string]Experiment, error)
	CreateExperiment(name string) (Experiment, error)
	GetOrCreateExperimentWithName(name string) (Experiment, error)
	GetExperiment(id string) (Experiment, error)
	URI() string
	UIURL() string
	// Returns (matching runs, next page token, error)
	SearchRuns(experimentIDs []string, filter string, orderBy []string, pageToken string) ([]Run, string, error)
}