Skip to content

Instantly share code, notes, and snippets.

@wbgalvao

wbgalvao/clean_mlflow_registered_models.py

Created February 13, 2025 15:01
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save wbgalvao/e24f27733efa2c907a09981e98763f90 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save wbgalvao/e24f27733efa2c907a09981e98763f90 to your computer and use it in GitHub Desktop.
Download ZIP
Deleted registered models in MLFlow
Raw
clean_mlflow_registered_models.py
import mlflow
client = mlflow.tracking.MlflowClient()
page_token = None
while True:
results = client.search_registered_models(page_token=page_token)
for model in results:
if "elasticity_regression" in model.name:
client.delete_registered_model(name=model.name)
print(f"deleted {model.name}")
if not results.token:
break
page_token = results.token
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment