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: