allegroai / clearml

ClearML - Auto-Magical CI/CD to streamline your AI workload. Experiment Management, Data Management, Pipeline, Orchestration, Scheduling & Serving in one MLOps/LLMOps solution
https://clear.ml/docs
Apache License 2.0
5.61k stars 651 forks source link

Add capability to archive/unarchive models from a model registry programmatically #1096

Open hariishaa opened 1 year ago

hariishaa commented 1 year ago

Proposal Summary

Capability to archive/unarchive models from a model registry programmatically.

Motivation

I would like to manage states of models in model registry from unpublished to published and then from published to archived and do all this actions from code. Currently only a function publish() is available in Model class, but function archive() is absent.

eugen-ajechiloae-clearml commented 1 year ago

Hi @hariishaa ! Thank you for the suggestion. We will soon add this functionality.

pollfly commented 1 year ago

Hey @hariishaa! v1.13.0 is now out, supporting archiving and unarchiving models using Model.archive() and Model.unarchive()

hariishaa commented 11 months ago

@pollfly this is good news! Thank you! But unfortunately I noticed that I am also missing the 'archived' attribute to determine the status of a model in a registry (whether it is archived or not).

ainoam commented 11 months ago

@hariishaa the archive status of a model is noted through the 'archived' system tag. Does this help?

hariishaa commented 11 months ago

@hariishaa the archive status of a model is noted through the 'archived' system tag. Does this help?

Yeah, that's a nice workaround, but it's not obvious from the docs. I am sure that it would be better to have 'archived' attribute for the models.

ainoam commented 11 months ago

Whole heartedly agree @hariishaa. We'll look into adding an explicit interface in a near version.