Struct ModelVersionControl 

pub struct ModelVersionControl { /* private fields */ }

Model version control

Implementations

impl ModelVersionControl

pub fn new() -> Self

pub fn initialize(&mut self) -> Result<(), MLError>

pub fn create_version( &mut self, model_id: &str, version: ModelVersion, ) -> Result<(), MLError>

Register a new version for a model. Returns an error if a version with the same version_id is already registered for that model.

pub fn get_version( &self, model_id: &str, version_id: &str, ) -> Option<&ModelVersion>

Get a specific version of a model by its version id.

pub fn list_versions(&self, model_id: &str) -> Vec<String>

List all version ids registered for a model.

pub fn create_branch( &mut self, branch_name: &str, from_version: &str, ) -> Result<(), MLError>

Create a branch starting from an existing version. The branch initially contains only the originating version.

pub fn get_branch(&self, branch_name: &str) -> Option<&Vec<String>>

Get the list of version ids in a branch.

pub fn tag_version( &mut self, version_id: &str, tag: &str, ) -> Result<(), MLError>

Tag a version. Multiple tags may be attached to the same version.

pub fn get_tags(&self, version_id: &str) -> Vec<String>

Get all tags attached to a version.

pub fn get_by_tag(&self, tag: &str) -> Vec<String>

Find all version ids that carry the given tag.