mlvern.utils package

Submodules

mlvern.utils.hashing module

mlvern.utils.hashing.hash_object(obj) → str[source]

mlvern.utils.registry module

mlvern.utils.registry.get_run_tags(mlvern_dir: str, run_id: str) → Dict[str, Any][source]

Get tags for a specific run.

Args:: mlvern_dir: Path to mlvern directory run_id: Run identifier
Returns:: Dict of tags, or empty dict if none exist

mlvern.utils.registry.init_registry(mlvern_dir, project_name)[source]

mlvern.utils.registry.list_models_registry(mlvern_dir: str) → Dict[str, Any][source]

List all registered models.

Args:: mlvern_dir: Path to mlvern directory
Returns:: Dict of model_id -> metadata

mlvern.utils.registry.load_registry(mlvern_dir)[source]

mlvern.utils.registry.register_model_metadata(mlvern_dir: str, model_id: str, metadata: Dict[str, Any]) → None[source]

Register model metadata in the registry.

Args:: mlvern_dir: Path to mlvern directory model_id: Unique model identifier metadata: Model metadata dict (class, hyperparameters,

source run, etc.)

mlvern.utils.registry.save_registry(mlvern_dir, registry)[source]

mlvern.utils.registry.search_runs_by_tag(mlvern_dir: str, tag_key: str, tag_value: Any) → list[source]

Search runs by tag key-value pair.

Args:: mlvern_dir: Path to mlvern directory tag_key: Tag key to search for tag_value: Tag value to match
Returns:: List of run IDs matching the tag

mlvern.utils.registry.tag_run(mlvern_dir: str, run_id: str, tags: Dict[str, Any]) → None[source]

Add or update tags on a run.

Args:: mlvern_dir: Path to mlvern directory run_id: Run identifier tags: Tags dict to merge with existing tags

Module contents

mlvern.utils.init_registry(mlvern_dir, project_name)[source]

mlvern.utils.load_registry(mlvern_dir)[source]

mlvern.utils.save_registry(mlvern_dir, registry)[source]