githubmirror/Made-With-ML

tests/code/utils.py

13 строк395 B

GokuMohandas

ML for Developers

26 июл 2023, 14:53

26 июл 2023, 14:53776a75b

import uuid

from madewithml.config import mlflow


def generate_experiment_name(prefix: str = "test") -> str:
    return f"{prefix}-{uuid.uuid4().hex[:8]}"


def delete_experiment(experiment_name: str) -> None:
    client = mlflow.tracking.MlflowClient()
    experiment_id = client.get_experiment_by_name(experiment_name).experiment_id
    client.delete_experiment(experiment_id=experiment_id)