Skip to content

fix: preserve model version tags in OSS bulk export#244

Open
0guban0v wants to merge 1 commit into
mlflow:masterfrom
0guban0v:fix/model-version-tags-lost-oss-export
Open

fix: preserve model version tags in OSS bulk export#244
0guban0v wants to merge 1 commit into
mlflow:masterfrom
0guban0v:fix/model-version-tags-lost-oss-export

Conversation

@0guban0v

@0guban0v 0guban0v commented May 11, 2026

Copy link
Copy Markdown

Summary

Fixes #240.

On PostgreSQL backends, search_model_versions (used by SearchModelVersionsIterator) does not return tags or aliases on model versions. The non-UC path in list_model_versions was returning raw search results directly, so version tags were silently dropped during export.

The UC path already handled this correctly by calling get_model_version per version. This PR applies the same approach to the non-UC path.

search_model_versions omits tags/aliases on PostgreSQL backends. The non-UC
path in list_model_versions was returning raw search results; now calls
get_model_version per version, matching the existing UC path.

@B-Step62 B-Step62 left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@0guban0v

Copy link
Copy Markdown
Author

@dbczumar could you complete review ?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Model version tags are lost in full-instance OSS migration (export-all/import-all)

2 participants