MLflow module was moved

dan/ocr/train.py

@@ -20,7 +20,7 @@ from dan.utils import MLflowNotInstalled
 if MLFLOW_AVAILABLE:
     import mlflow
 
-    from dan.mlflow import make_mlflow_request, start_mlflow_run
+    from dan.ocr.mlflow import make_mlflow_request, start_mlflow_run
 
 
 logger = logging.getLogger(__name__)