set mlflow env var to None

mosaicml · Nov 19, 2024 · f00bded · f00bded
1 parent 78ba60c
commit f00bded
Showing 1 changed file with 6 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,12 @@
 DEFAULT_PATH = TEST_PATH_FORMAT.format(experiment_id=EXPERIMENT_ID, run_id=RUN_ID)
 
 
+@pytest.fixture(autouse=True)
+def setup_mlflow_tracking(monkeypatch):
+    mlflow = pytest.importorskip('mlflow')
+    monkeypatch.setattr(mlflow.environment_variables, 'MLFLOW_TRACKING_URI', None)
+
+
 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)