minor

config/config.py

@@ -5,7 +5,6 @@ import logging.config
 import sys
 from pathlib import Path
 
-import mlflow
 from rich.logging import RichHandler
 
 # Base and Config Directories
@@ -54,9 +53,6 @@ assert TRAIN_SIZE + VAL_SIZE + TEST_SIZE == 1
 with open(Path(CONFIG_DIR, "heteronyms.json")) as fp:
     HETERONYMS = json.load(fp)
 
-# MLFlow model registry
-mlflow.set_tracking_uri("file://" + str(RUN_REGISTRY.absolute()))
-
 # Logger
 logging_config = {
     "version": 1,

yomikata/main.py

@@ -16,6 +16,11 @@ from config.config import logger
 from yomikata import utils
 from yomikata.dbert import dBert
 
+
+# MLFlow model registry
+mlflow.set_tracking_uri("file://" + str(config.RUN_REGISTRY.absolute()))
+
+
 warnings.filterwarnings("ignore")