src/utils/mlflow_utils.py

import mlflow


def setup_mlflow(experiment_name: str):
    mlflow.set_tracking_uri("sqlite:///mlflow.db")
    mlflow.set_experiment(experiment_name)