entrenar 0.7.12

Training & Optimization library with autograd, LoRA, quantization, and model merging
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

//! Model Registry with Staging Workflows (MLOPS-008)
//!
//! Kanban-style model lifecycle management.
//!
//! # Toyota Way: (Kanban)
//!
//! Visual workflow stages for model promotion with pull-based progression.
//! Models flow: None -> Development -> Staging -> Production -> Archived
//!
//! # Example
//!
//! ```ignore
//! use entrenar::storage::registry::{ModelRegistry, ModelStage, InMemoryRegistry};
//!
//! let mut registry = InMemoryRegistry::new();
//! registry.register_model("llama-7b-finetuned", "path/to/model.safetensors")?;
//! registry.transition_stage("llama-7b-finetuned", 1, ModelStage::Staging, Some("alice"))?;
//! ```

mod comparison;
mod error;
mod memory;
mod policy;
mod stage;
mod traits;
mod transition;
mod version;

// Re-export all public types for API compatibility
pub use comparison::{Comparison, MetricRequirement, VersionComparison};
pub use error::{RegistryError, Result};
pub use memory::InMemoryRegistry;
pub use policy::{PolicyCheckResult, PromotionPolicy};
pub use stage::ModelStage;
pub use traits::ModelRegistry;
pub use transition::StageTransition;
pub use version::ModelVersion;