From e9f6b6d5d3610cbda8e15f3c13148097b3618312 Mon Sep 17 00:00:00 2001
From: Yoann Schneider <yschneider@teklia.com>
Date: Fri, 10 Feb 2023 11:05:10 +0100
Subject: [PATCH] do not force mlflow
---
dan/ocr/document/train.py | 8 ++++++++
1 file changed, 8 insertions(+)
diff --git a/dan/ocr/document/train.py b/dan/ocr/document/train.py
index 4b005ec4..93829fe0 100644
--- a/dan/ocr/document/train.py
+++ b/dan/ocr/document/train.py
@@ -28,6 +28,14 @@ try:
except ImportError:
MLFLOW = False
+try:
+ import mlflow
+
+ MLFLOW = True
+ logger.info("MLflow Logging available.")
+except ImportError:
+ MLFLOW = False
+
logger = logging.getLogger(__name__)
--
GitLab