From 00d340f39fbaf234ce8fa590206dc8fe75cea24d Mon Sep 17 00:00:00 2001
From: Yoann Schneider <yschneider@teklia.com>
Date: Fri, 10 Feb 2023 10:49:30 +0000
Subject: [PATCH] check if mlflow is available at runtime
---
dan/manager/training.py | 5 +++++
dan/ocr/document/train.py | 7 -------
dan/utils.py | 3 +++
3 files changed, 8 insertions(+), 7 deletions(-)
diff --git a/dan/manager/training.py b/dan/manager/training.py
index 7b565c41..921f1c8a 100644
--- a/dan/manager/training.py
+++ b/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
diff --git a/dan/ocr/document/train.py b/dan/ocr/document/train.py
index 93829fe0..7e2729f1 100644
--- a/dan/ocr/document/train.py
+++ b/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__)
diff --git a/dan/utils.py b/dan/utils.py
index 50f7311d..e11c2e85 100644
--- a/dan/utils.py
+++ b/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
--
GitLab