6866ba9366
Update queue and storage systems for scheduler integration: - Queue backend with scheduler coordination - Filesystem queue with batch operations - Deduplication with tenant-aware keys - Storage layer with audit logging hooks - Domain models (Task, Events, Errors) with scheduler fields - Database layer with tenant isolation - Dataset storage with integrity checks
30 lines
1 KiB
Go
30 lines
1 KiB
Go
package domain
|
|
|
|
// TrackingConfig specifies experiment tracking tools to enable for a task.
|
|
type TrackingConfig struct {
|
|
MLflow *MLflowTrackingConfig `json:"mlflow,omitempty"`
|
|
TensorBoard *TensorBoardTrackingConfig `json:"tensorboard,omitempty"`
|
|
Wandb *WandbTrackingConfig `json:"wandb,omitempty"`
|
|
}
|
|
|
|
// MLflowTrackingConfig controls MLflow integration.
|
|
type MLflowTrackingConfig struct {
|
|
Mode string `json:"mode,omitempty"`
|
|
TrackingURI string `json:"tracking_uri,omitempty"`
|
|
Enabled bool `json:"enabled"`
|
|
}
|
|
|
|
// TensorBoardTrackingConfig controls TensorBoard integration.
|
|
type TensorBoardTrackingConfig struct {
|
|
Mode string `json:"mode,omitempty"`
|
|
Enabled bool `json:"enabled"`
|
|
}
|
|
|
|
// WandbTrackingConfig controls Weights & Biases integration.
|
|
type WandbTrackingConfig struct {
|
|
Mode string `json:"mode,omitempty"`
|
|
APIKey string `json:"api_key,omitempty"`
|
|
Project string `json:"project,omitempty"`
|
|
Entity string `json:"entity,omitempty"`
|
|
Enabled bool `json:"enabled"`
|
|
}
|