From e01c3e8bdbdc75312cdd273928e8ab811a66b28a Mon Sep 17 00:00:00 2001 From: Lehui Liu Date: Tue, 21 Oct 2025 15:30:41 -0700 Subject: [PATCH 1/3] fix a doc off Signed-off-by: Lehui Liu --- doc/source/train/user-guides/experiment-tracking.rst | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/doc/source/train/user-guides/experiment-tracking.rst b/doc/source/train/user-guides/experiment-tracking.rst index 424aff50f798..adf0f1770b7c 100644 --- a/doc/source/train/user-guides/experiment-tracking.rst +++ b/doc/source/train/user-guides/experiment-tracking.rst @@ -182,8 +182,10 @@ If applicable, make sure that you properly set up credentials on each training w .. testcode:: :skipif: True - - mlflow.start_run(tracking_uri="file:some_shared_storage_path/mlruns") + + mlflow.set_tracking_uri("file://some_shared_storage_path/mlruns") + mlflow.set_experiment("my_experiment") + mlflow.start_run() **Remote, hosted by Databricks** From 7e2037f0833f33a1b9128a6f786ec017004a260b Mon Sep 17 00:00:00 2001 From: Lehui Liu Date: Tue, 21 Oct 2025 16:15:55 -0700 Subject: [PATCH 2/3] add uri= Signed-off-by: Lehui Liu --- doc/source/train/user-guides/experiment-tracking.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/doc/source/train/user-guides/experiment-tracking.rst b/doc/source/train/user-guides/experiment-tracking.rst index adf0f1770b7c..b765eabf471a 100644 --- a/doc/source/train/user-guides/experiment-tracking.rst +++ b/doc/source/train/user-guides/experiment-tracking.rst @@ -183,7 +183,7 @@ If applicable, make sure that you properly set up credentials on each training w .. testcode:: :skipif: True - mlflow.set_tracking_uri("file://some_shared_storage_path/mlruns") + mlflow.set_tracking_uri(uri="file://some_shared_storage_path/mlruns") mlflow.set_experiment("my_experiment") mlflow.start_run() From ef832cf9753e0f86ce9ed01b553855ddee2476d9 Mon Sep 17 00:00:00 2001 From: Lehui Liu Date: Tue, 21 Oct 2025 17:06:00 -0700 Subject: [PATCH 3/3] remove setup experiment Signed-off-by: Lehui Liu --- doc/source/train/user-guides/experiment-tracking.rst | 1 - 1 file changed, 1 deletion(-) diff --git a/doc/source/train/user-guides/experiment-tracking.rst b/doc/source/train/user-guides/experiment-tracking.rst index b765eabf471a..4e38d3dd6373 100644 --- a/doc/source/train/user-guides/experiment-tracking.rst +++ b/doc/source/train/user-guides/experiment-tracking.rst @@ -184,7 +184,6 @@ If applicable, make sure that you properly set up credentials on each training w :skipif: True mlflow.set_tracking_uri(uri="file://some_shared_storage_path/mlruns") - mlflow.set_experiment("my_experiment") mlflow.start_run() **Remote, hosted by Databricks**