From ae32f5d994085bdac0f6fcbbe00c00468fd5e7a8 Mon Sep 17 00:00:00 2001 From: denver Date: Tue, 18 Mar 2025 17:22:38 -0500 Subject: [PATCH] test: add explicit mlflow env setting in test_config.py --- graphdoc/tests/test_config.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/graphdoc/tests/test_config.py b/graphdoc/tests/test_config.py index 2d23787..3ef7458 100644 --- a/graphdoc/tests/test_config.py +++ b/graphdoc/tests/test_config.py @@ -4,6 +4,7 @@ import logging # system packages +import os from pathlib import Path # external packages @@ -40,6 +41,12 @@ BASE_DIR = Path(__file__).resolve().parent.parent SCHEMA_DIR = BASE_DIR / "tests" / "assets" / "schemas" CONFIG_DIR = BASE_DIR / "tests" / "assets" / "configs" +MLRUNS_DIR = BASE_DIR / "tests" / "assets" / "mlruns" + +# set the environment variables +os.environ["MLFLOW_TRACKING_URI"] = str(MLRUNS_DIR) +os.environ["MLFLOW_TRACKING_USERNAME"] = "admin" +os.environ["MLFLOW_TRACKING_PASSWORD"] = "password" class TestConfig: