From d3f3414f6a010218783a297c472c4fb0c4f94df8 Mon Sep 17 00:00:00 2001
From: Yoann Schneider <yschneider@teklia.com>
Date: Fri, 10 Feb 2023 11:50:05 +0100
Subject: [PATCH] fix lint
---
dan/manager/training.py | 2 ++
dan/mlflow.py | 1 -
dan/utils.py | 1 +
3 files changed, 3 insertions(+), 1 deletion(-)
diff --git a/dan/manager/training.py b/dan/manager/training.py
index d7b4897c..7b565c41 100644
--- a/dan/manager/training.py
+++ b/dan/manager/training.py
@@ -23,8 +23,10 @@ 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
diff --git a/dan/mlflow.py b/dan/mlflow.py
index 20b3436e..063c2b91 100644
--- a/dan/mlflow.py
+++ b/dan/mlflow.py
@@ -8,7 +8,6 @@ from mlflow.exceptions import MlflowException
from dan import logger
-
def setup_environment(config: dict):
"""
Get the necessary variables from the config file and put them in the environment variables
diff --git a/dan/utils.py b/dan/utils.py
index ad8d3ca4..50f7311d 100644
--- a/dan/utils.py
+++ b/dan/utils.py
@@ -23,6 +23,7 @@ class MLflowNotInstalled(Exception):
Raised when MLflow logging was requested but the module was not installed
"""
+
def randint(low, high):
"""
call torch.randint to preserve random among dataloader workers
--
GitLab