def _get_or_start_run(run_name: Optional[str]) -> "ActiveRun":
    """The function of MLflow. Gets the active run and gives it a name.
    If active run does not exist, starts a new one.

    Args:
        run_name: Name of the run

    Returns:
        ActiveRun
    """
    active_run = mlflow.active_run()
    if active_run:
        mlflow.set_tag("mlflow.runName", run_name)
        return active_run
    return mlflow.start_run(run_name=run_name)
The function of MLflow. Gets the active run and gives it a name.
    If active run does not exist, starts a new one.

    Args:
        run_name: Name of the run

    Returns:
        ActiveRun
    
catalyst/loggers/mlflow.py