Struct InMemoryRegistry 

pub struct InMemoryRegistry { /* private fields */ }

In-memory model registry for testing

Implementations

impl InMemoryRegistry

pub fn new() -> Self

Create a new in-memory registry

pub fn enable_auto_rollback( &mut self, model: &str, metric: &str, threshold: f64, )

Enable auto-rollback for a model

pub fn check_rollback(&self, model: &str, current_metric: f64) -> bool

Check if rollback is needed based on metrics

Trait Implementations

impl Debug for InMemoryRegistry

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for InMemoryRegistry

fn default() -> InMemoryRegistry

Returns the “default value” for a type.

impl ModelRegistry for InMemoryRegistry

fn register_model( &mut self, name: &str, artifact_uri: &str, ) -> Result<ModelVersion>

Register a new model version

fn get_model(&self, name: &str, version: u32) -> Result<ModelVersion>

Get a model version

fn get_latest(&self, name: &str) -> Result<ModelVersion>

Get latest version of a model

fn get_latest_by_stage( &self, name: &str, stage: ModelStage, ) -> Option<ModelVersion>

Get latest version at a specific stage

fn list_versions(&self, name: &str) -> Result<Vec<ModelVersion>>

List all versions of a model

fn transition_stage( &mut self, name: &str, version: u32, target_stage: ModelStage, user: Option<&str>, ) -> Result<()>

Transition model to new stage

fn compare_versions( &self, name: &str, v1: u32, v2: u32, ) -> Result<VersionComparison>

Compare two versions

fn log_metrics( &mut self, name: &str, version: u32, metrics: HashMap<String, f64>, ) -> Result<()>

Log metrics for a model version

fn get_transition_history(&self, name: &str) -> Result<Vec<StageTransition>>

Get transition history for a model

fn set_policy(&mut self, policy: PromotionPolicy)

Set promotion policy for a stage

fn get_policy(&self, stage: ModelStage) -> Option<&PromotionPolicy>

Get promotion policy for a stage

fn can_promote( &self, name: &str, version: u32, target_stage: ModelStage, approvals: u32, ) -> Result<PolicyCheckResult>

Check if model can be promoted (with policy check)