mlflow# fkat.utils.mlflow.broadcast_mlflow_run_id(mlflow: MLFlowLogger, trainer: Trainer) → None[source]# Broadcast mlflow run_id from rank0 to all ranks and setup the mlflow logger. We assume PTL mlflow logger is only initialized with a run_id on rank0 via logger.experiment. fkat.utils.mlflow.mlflow_logger(trainer: Trainer) → MLFlowLogger | None[source]# Returns MLFlowLogger from trainer as constructed by PyTorch Lightning.