set tracking

mosaicml · Nov 19, 2024 · 1052a16 · 1052a16
1 parent 17ca05a
commit 1052a16
Showing 1 changed file with 3 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
@@ -138,6 +138,9 @@ def test_init_with_no_experiment_and_no_run(monkeypatch):
     assert store.experiment_id == EXPERIMENT_ID
     assert store.run_id == RUN_ID
 
+@pytest.fixture(autouse=True)
+def set_mlflow_tracking_uri(monkeypatch):
+    monkeypatch.setenv('MLFLOW_TRACKING_URI', 'databricks')
 
 @pytest.fixture()
 def mlflow_object_store(monkeypatch):