Skip to main content

Module storage

Module storage 

Source
Expand description

Experiment Storage Module (ENT-001)

Provides the ExperimentStorage trait and backends for persisting experiment tracking data.

§Backends

  • TruenoBackend: Production backend using trueno-db (feature: “monitor”)
  • InMemoryStorage: In-memory backend for testing and fuzzing

§Example

use entrenar::storage::{ExperimentStorage, InMemoryStorage, RunStatus};

let mut storage = InMemoryStorage::new();
let exp_id = storage.create_experiment("my-experiment", None).expect("create experiment");
let run_id = storage.create_run(&exp_id).expect("create run");
storage.start_run(&run_id).expect("start run");
storage.log_metric(&run_id, "loss", 0, 0.5).expect("log metric");
storage.complete_run(&run_id, RunStatus::Success).expect("complete run");

Re-exports§

pub use cloud::ArtifactBackend;
pub use cloud::ArtifactMetadata;
pub use cloud::AzureConfig;
pub use cloud::BackendConfig;
pub use cloud::CloudError;
pub use cloud::GCSConfig;
pub use cloud::InMemoryBackend;
pub use cloud::LocalBackend;
pub use cloud::MockS3Backend;
pub use cloud::S3Config;
pub use memory::InMemoryStorage;
pub use preflight::CheckMetadata;
pub use preflight::CheckResult;
pub use preflight::CheckType;
pub use preflight::Preflight;
pub use preflight::PreflightCheck;
pub use preflight::PreflightContext;
pub use preflight::PreflightError;
pub use preflight::PreflightResults;
pub use registry::Comparison;
pub use registry::InMemoryRegistry;
pub use registry::MetricRequirement;
pub use registry::ModelRegistry;
pub use registry::ModelStage;
pub use registry::ModelVersion;
pub use registry::PolicyCheckResult;
pub use registry::PromotionPolicy;
pub use registry::RegistryError;
pub use registry::StageTransition;
pub use registry::VersionComparison;
pub use sqlite::ArtifactRef;
pub use sqlite::Experiment;
pub use sqlite::FilterOp;
pub use sqlite::ParamFilter;
pub use sqlite::ParameterValue;
pub use sqlite::Run;
pub use sqlite::SqliteBackend;
pub use trueno::TruenoBackend;

Modules§

cloud
Cloud Storage Backends (#72)
memory
In-Memory Storage Backend
preflight
Pre-flight Validation System (Jidoka)
registry
Model Registry with Staging Workflows (MLOPS-008)
sqlite
SQLite Backend for Experiment Storage (MLOPS-001)
trueno
TruenoDB Storage Backend (ENT-001)

Structs§

MetricPoint
A single metric data point

Enums§

RunStatus
Status of a run
StorageError
Storage errors

Traits§

ExperimentStorage
Trait for experiment storage backends

Type Aliases§

Result
Result type for storage operations