image-match/python/py/Lib/site-packages/ray/tune/logger/mlflow.py

from ray.air.integrations.mlflow import MLflowLoggerCallback

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