set up tracking

mosaicml · Nov 19, 2024 · 0f8ae76 · 0f8ae76
1 parent a9d8c12
commit 0f8ae76
Showing 1 changed file with 5 additions and 0 deletions.
diff --git 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)