[DEPRECATIONS] Support mlflow version 3.x #9102

omermaim · 2025-12-22T12:59:22Z

…n new version of mlflow

📝 Description

upgraded mlflow version to 3.0.0

🛠️ Changes Made

changed logic in model retrievel due to new mlflow objects. the models are no longer in list_artifacts function

✅ Checklist

I updated the documentation (if applicable)
I have tested the changes in this PR
I confirmed whether my changes are covered by system tests
- If yes, I ran all relevant system tests and ensured they passed before submitting this PR
- I updated existing system tests and/or added new ones if needed to cover my changes
If I introduced a deprecation:
- I followed the Deprecation Guidelines
- I updated the relevant Jira ticket for documentation

🧪 Testing

ran alll tracker unit tests that is the only package that has this library.

🔗 References

Ticket link: https://iguazio.atlassian.net/browse/ML-11537?actionerId=712020%3A4a32fc43-a2c3-4d13-b5d2-7e4c9f30172e&sourceType=assign
Design docs links:
External links:

🚨 Breaking Changes?

Yes (explain below)
No

🔍️ Additional Notes

removes mlflow from the mlrun/mlrun image.

…n new version of mlflow

theSaarco

Our general approach should be to not install the mlflow package unless specifically asked for. This means that we want to remove this package from the complete extra, and install it only if the user asks for the mlflow extra. This will reduce our exposure to the package and its possible vulns and dependencies.
I've added some comments explaining the exact changes that are needed to facilitate this.
@liranbg, @guy1992l - FYI.

theSaarco · 2025-12-22T14:20:53Z

dependencies.py

        ],
        "redis": ["redis~=4.3"],
-        "mlflow": ["mlflow~=2.22"],
+        "mlflow": ["mlflow~=3.0.0"],


Having discussed this in the release-tracking meeting, we want to remove the mlflow requirement from the complete extra. This means that this line should be removed from here, and added below (see L84 and further) so that when users ask for mlrun["mlflow"] they get just mlflow, and to install complete + mlflow they'll need to do install mlrun["complete, mlflow"].

dependencies.py

theSaarco · 2025-12-22T14:27:50Z

extras-requirements.txt

 redis~=4.3
 graphviz~=0.20.0
-mlflow~=2.22
+mlflow~=3.0.0


As mentioned above, this also needs to be removed so we don't install this package by default in our images.

jond01 · 2025-12-23T08:47:24Z

🚨 Breaking Changes?

Yes (explain below)

[ x] No

@omermaim - please mark this as a breaking change :) Also in Jira

gilad-shaham · 2025-12-23T11:54:42Z

dependencies.py

        {
            "dev-postgres": ["pytest-mock-resources[postgres]~=2.12"],
            "kfp18": ["mlrun_pipelines_kfp_v1_8[kfp]~=0.6.0"],
+            "mlflow": ["mlflow~=3.0.0"],


If we went all the way up to 3, why ~=3.0.0 and not 3.8.0? Does something break for us between 3.0 and 3.8?

in the ticket it said to move to version 3.0.0. Anyway this change should support later versions as well.

theSaarco

LGTM

guy1992l

VERY GOOD JOB :)

guy1992l · 2025-12-23T13:01:24Z

dependencies.py

        {
            "dev-postgres": ["pytest-mock-resources[postgres]~=2.12"],
            "kfp18": ["mlrun_pipelines_kfp_v1_8[kfp]~=0.6.0"],
+            "mlflow": ["mlflow~=3.0.0"],


in the ticket it said to move to version 3.0.0. Anyway this change should support later versions as well.

guy1992l · 2025-12-23T13:03:18Z

dependencies.py

        {
            "dev-postgres": ["pytest-mock-resources[postgres]~=2.12"],
            "kfp18": ["mlrun_pipelines_kfp_v1_8[kfp]~=0.6.0"],
+            "mlflow": ["mlflow~=3.0.0"],


@gilad-shaham @theSaarco

mlflow~=3.0.0 is very strict. Question is should we strict it as security violations do not exist for it or should we move to mlflow~=3.0. The changes in their SDK and API should be fine.

guy1992l · 2025-12-23T13:03:43Z

dockerfiles/test/requirements.txt

@@ -1 +1,2 @@
 mlrun-pipelines-kfp-v1-8[kfp]~=0.6.0
+mlflow~=3.0.0


Same here...

guy1992l · 2025-12-23T13:05:39Z

tests/test_requirements.py

        "aioresponses": {"~=0.7"},
        "testcontainers[k3s]": {"~=4.10.0"},
        "scikit-learn": {"~=1.5.2"},
+        "mlflow": {"~=3.0.0"},


guy1992l · 2025-12-23T13:06:55Z

mlrun/track/trackers/mlflow_tracker.py

+        local_model_path = mlflow.artifacts.download_artifacts(
+            artifact_uri=str(model_uri)
+        )
+        model_path = pathlib.Path(local_model_path)


Please keep the headline comments.

Suggested change

local_model_path = mlflow.artifacts.download_artifacts(

artifact_uri=str(model_uri)

)

model_path = pathlib.Path(local_model_path)

# Download the model and set the path to local path:

local_model_path = mlflow.artifacts.download_artifacts(

artifact_uri=str(model_uri)

)

model_path = pathlib.Path(local_model_path)

guy1992l · 2025-12-23T13:09:58Z

tests/track/platform_trackers/test_mlflow_tracker.py

+            filter_string=f"source_run_id = '{mlflow_run.info.run_id}'"
+        )
+
+        model_uri = logged_models.iloc[0].artifact_location


add output_format="list" to search_logged_models to get a list instead of DataFrame.

- change logged models ouput to be list

…low version

mlflow version 3.0.0, changed logic in tracker due to object change i…

e15d0d4

…n new version of mlflow

omermaim requested review from a team and liranbg as code owners December 22, 2025 12:59

github-actions bot added PR title format area/sdk area/tests labels Dec 22, 2025

omermaim changed the title ~~mlflow version 3.0.0, changed logic in tracker due to object change i…~~ [DEPRECATIONS] mlflow version 3.0.0 Dec 22, 2025

github-actions bot removed the PR title format label Dec 22, 2025

Merge branch 'development' into ML-11565

2d93382

theSaarco requested changes Dec 22, 2025

View reviewed changes

omermaim added 3 commits December 23, 2025 10:17

removed mlflow from the complete extra

c4cf459

Merge branch 'development' into ML-11565

facc542

changed lock requirments after omitting mlflow from [complete] extra

debab0b

omermaim added 4 commits December 23, 2025 11:26

lint

462b9b5

Merge branch 'development' into ML-11565

99c5770

add mlflow to test's requirements.txt for ci/cd testing

437801b

add mlflow to test's locked-requirements.txt.txt for ci/cd testing

95f22e2

gilad-shaham reviewed Dec 23, 2025

View reviewed changes

theSaarco approved these changes Dec 23, 2025

View reviewed changes

omermaim added 3 commits December 23, 2025 14:42

ignore mlflow in test_requirements.py, only used in test image

4e5549d

Merge branch 'development' into ML-11565

1d64120

lint

058fe54

guy1992l requested changes Dec 23, 2025

View reviewed changes

omermaim added 3 commits December 23, 2025 15:26

- change version from 3.0.0 to 3.0 to avoid strictness

fa3cba0

- change logged models ouput to be list

lint + changed locked-requirements.txt of dockerfiles/test to new mlf…

9cd81aa

…low version

in test_requirement_specifiers_convention no need for mlflow

a3bac93

guy1992l changed the title ~~[DEPRECATIONS] mlflow version 3.0.0~~ [DEPRECATIONS] Support mlflow version 3.x Dec 23, 2025

guy1992l approved these changes Dec 23, 2025

View reviewed changes

liranbg merged commit 881c3cf into mlrun:development Dec 23, 2025
14 checks passed

omermaim deleted the ML-11565 branch December 24, 2025 08:15

		@@ -1 +1,2 @@
		mlrun-pipelines-kfp-v1-8[kfp]~=0.6.0
		mlflow~=3.0.0

[DEPRECATIONS] Support mlflow version 3.x #9102

[DEPRECATIONS] Support mlflow version 3.x #9102

Uh oh!

Conversation

omermaim commented Dec 22, 2025 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

📝 Description

🛠️ Changes Made

✅ Checklist

🧪 Testing

🔗 References

🚨 Breaking Changes?

🔍️ Additional Notes

Uh oh!

theSaarco left a comment

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Uh oh!

Choose a reason for hiding this comment

Uh oh!

jond01 commented Dec 23, 2025

🚨 Breaking Changes?

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

theSaarco left a comment

Choose a reason for hiding this comment

Uh oh!

guy1992l left a comment

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

6 participants

omermaim commented Dec 22, 2025 •

edited

Loading