from ray.air.integrations.mlflow import MLflowLoggerCallback

MLflowLoggerCallback.__module__ = "ray.tune.logger.mlflow"