This is an automated email from the ASF dual-hosted git repository.
gaogaotiantian pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/spark.git
The following commit(s) were added to refs/heads/master by this push:
new 21fee7d70fd5 [SPARK-57239][PYTHON][TEST] Use sqlite uri for mlflow
model
21fee7d70fd5 is described below
commit 21fee7d70fd59f4fbe23ff3b155a2aa97be6e967
Author: Tian Gao <[email protected]>
AuthorDate: Tue Jun 2 19:03:49 2026 -0700
[SPARK-57239][PYTHON][TEST] Use sqlite uri for mlflow model
### What changes were proposed in this pull request?
Use sqlite instead of file for mlflow doctest.
### Why are the changes needed?
Pandas CI is failing -
https://github.com/manhha2502/spark/actions/runs/26841118356/job/79154335111
The latest `mlflow` does not allow the usage of file anymore - it will
raise an exception.
### Does this PR introduce _any_ user-facing change?
No.
### How was this patch tested?
Locally passed, pending CI.
### Was this patch authored or co-authored using generative AI tooling?
No.
Closes #56290 from gaogaotiantian/use-sqlite-uri.
Authored-by: Tian Gao <[email protected]>
Signed-off-by: Tian Gao <[email protected]>
---
python/pyspark/pandas/mlflow.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/python/pyspark/pandas/mlflow.py b/python/pyspark/pandas/mlflow.py
index f15ff83b11cc..06988a2871c6 100644
--- a/python/pyspark/pandas/mlflow.py
+++ b/python/pyspark/pandas/mlflow.py
@@ -142,7 +142,7 @@ def load_model(
>>> import mlflow.sklearn
>>> from tempfile import mkdtemp
>>> d = mkdtemp("pandas_on_spark_mlflow")
- >>> set_tracking_uri("file:%s"%d)
+ >>> set_tracking_uri(f"sqlite:///{d}/mlflow.db")
>>> client = MlflowClient()
>>> exp_id = mlflow.create_experiment("my_experiment")
>>> exp = mlflow.set_experiment("my_experiment")
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
