check if mlflow is available at runtime

dan/manager/training.py

@@ -23,6 +23,11 @@ from tqdm import tqdm
 from dan.manager.metrics import MetricManager
 from dan.ocr.utils import LM_ind_to_str
 from dan.schedulers import DropoutScheduler
+try:
+    import mlflow
+    from dan.mlflow import logging_metrics, logging_tags_metrics
+except ImportError:
+    pass
 
 try:
     import mlflow

dan/ocr/document/train.py

@@ -28,13 +28,6 @@ try:
 except ImportError:
     MLFLOW = False
 
-try:
-    import mlflow
-
-    MLFLOW = True
-    logger.info("MLflow Logging available.")
-except ImportError:
-    MLFLOW = False
 
 logger = logging.getLogger(__name__)
 

dan/utils.py

@@ -23,7 +23,10 @@ class MLflowNotInstalled(Exception):
     Raised when MLflow logging was requested but the module was not installed
     """
 
+<<<<<<< HEAD
 
+=======
+>>>>>>> bf9cf44 (check if mlflow is available at runtime)
 def randint(low, high):
     """
     call torch.randint to preserve random among dataloader workers