fix lint

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

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

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