diff --git a/tests/utils/object_store/test_mlflow_object_store.py b/tests/utils/object_store/test_mlflow_object_store.py index fd802048bc..20280cdcde 100644 --- a/tests/utils/object_store/test_mlflow_object_store.py +++ b/tests/utils/object_store/test_mlflow_object_store.py @@ -19,6 +19,11 @@ DEFAULT_PATH = TEST_PATH_FORMAT.format(experiment_id=EXPERIMENT_ID, run_id=RUN_ID) +@pytest.fixture(autouse=True) +def setup_mlflow_tracking(monkeypatch): + monkeypatch.setenv('MLFLOW_TRACKING_URI', 'databricks') + + def test_parse_dbfs_path(): full_artifact_path = DEFAULT_PATH + ARTIFACT_PATH assert MLFlowObjectStore.parse_dbfs_path(full_artifact_path) == (EXPERIMENT_ID, RUN_ID, ARTIFACT_PATH)