mlflow: delete all versions of a model

 # Delete all versions of a model:

client = mlflow.tracking.MlflowClient()
model_name = "model_name"

# get the range manually
for i in range(35,43):
    # convert first to archived stage else can't be deleted
    client.transition_model_version_stage(
        name=model_name,
        version=i,
        stage="archived"
    )
    # delete the version
    client.delete_model_version(model_name, i)

Comments