feat: Add MLflow integration support to Feast operator by Vperiodt · Pull Request #6611 · feast-dev/feast

Vperiodt · 2026-07-17T12:42:18Z

What this PR does / why we need it:

Adds optional spec.mlflow.uiUrl and wires it into feature_store.yaml as ui_url
Auto-discovers tracking URI (status.address.url) and UI URL (status.url) from the cluster MLflow CR
Feast UI lineage links use MlflowConfig.get_ui_url() (falls back to tracking URI for local/dev)

Which issue(s) this PR fixes:

Checks

I've made sure the tests are passing.
My commits are signed off (git commit -s)
My PR title follows conventional commits format

Testing Strategy

Unit tests
Integration tests
Manual tests
Testing is not required for this change

Misc

codecov-commenter · 2026-07-17T12:46:25Z

⚠️ Please install the to ensure uploads and comments are reliably processed by Codecov.

Codecov Report

❌ Patch coverage is 0% with 10 lines in your changes missing coverage. Please review.
✅ Project coverage is 46.80%. Comparing base (b8dfcb0) to head (098fcbd).

Files with missing lines	Patch %	Lines
sdk/python/feast/mlflow_integration/config.py	0.00%	8 Missing ⚠️
sdk/python/feast/ui_server.py	0.00%	2 Missing ⚠️
❗ Your organization needs to install the Codecov GitHub app to enable full functionality.

Additional details and impacted files

@@            Coverage Diff             @@
##           master    #6611      +/-   ##
==========================================
- Coverage   46.81%   46.80%   -0.01%     
==========================================
  Files         415      415              
  Lines       50399    50407       +8     
  Branches     7214     7215       +1     
==========================================
  Hits        23592    23592              
- Misses      25157    25165       +8     
  Partials     1650     1650

Flag	Coverage Δ
go-feature-server	`30.58% <ø> (ø)`
python-unit	`48.12% <0.00%> (-0.01%)`	⬇️

Files with missing lines	Coverage Δ
sdk/python/feast/ui_server.py	`24.64% <0.00%> (ø)`
sdk/python/feast/mlflow_integration/config.py	`0.00% <0.00%> (ø)`

Continue to review full report in Codecov by Harness.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update b8dfcb0...098fcbd. Read the comment docs.

🚀 New features to boost your workflow:

❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.
📦 JS Bundle Analysis: Save yourself from yourself by tracking and limiting bundle sizes in JS merges.

ntkathole · 2026-07-30T06:59:28Z

+		needsDiscovery := cr.Spec.Mlflow.TrackingUri == nil || cr.Spec.Mlflow.UiUrl == nil
+		if needsDiscovery && feast.Handler.Client != nil {
+			if discovered, ok := DiscoverMlflow(feast.Handler.Context, feast.Handler.Client); ok {
+				if cr.Status.Applied.Mlflow.TrackingUri == nil {


Suggested change

if cr.Status.Applied.Mlflow.TrackingUri == nil {

if cr.Status.Applied.Mlflow != nil && cr.Status.Applied.Mlflow.TrackingUri == nil {

ntkathole · 2026-07-30T07:02:25Z

+			return condStatus == "True"
+		}
+	}
+	return true


should return false, Returning true here could cause the operator to use a partially-initialized MLflow instance.

ntkathole · 2026-07-30T07:03:23Z

+		return
+	}
+	mlflowEnv := []corev1.EnvVar{
+		{Name: "MLFLOW_TRACKING_AUTH", Value: "kubernetes-namespaced"},


shouldn't it be configurable and not hard-coded

ntkathole · 2026-07-30T07:05:15Z

+		rb.RoleRef = rbacv1.RoleRef{
+			APIGroup: "rbac.authorization.k8s.io",
+			Kind:     "ClusterRole",
+			Name:     "mlflow-integration",


Does the MLflow operator provide this role ? I don't see logic to create

ntkathole · 2026-07-30T07:06:33Z

+	EntityDfMaxRows     *int32                 `yaml:"entity_df_max_rows,omitempty"`
+	LogOperations       *bool                  `yaml:"log_operations,omitempty"`
+	OpsExperimentSuffix *string                `yaml:"ops_experiment_suffix,omitempty"`
+	ExtraConfig         map[string]interface{} `yaml:",inline,omitempty"`


If a user puts tracking_uri or enabled in ExtraConfig, it would create conflicting/duplicate YAML keys. A kubebuilder validation rule or runtime check should prevent this.

ntkathole · 2026-07-30T07:08:59Z

+func (r *FeatureStoreReconciler) mapMlflowToFeastRequests(ctx context.Context, _ client.Object) []reconcile.Request {
+	logger := log.FromContext(ctx)
+	var feastList feastdevv1.FeatureStoreList
+	if err := r.List(ctx, &feastList, client.InNamespace("")); err != nil {


consider improving client.InNamespace("")); (rate-limiting / debounce strategy) lists FeatureStores across the entire cluster on every MLflow CR change. On large clusters this could produce significant API server load.

ntkathole · 2026-08-11T13:15:14Z

@Vperiodt get_mlflow_feature_models is not fixed, still uses tracking_uri

https://github.com/Vperiodt/feast/blob/86691c91b273346f160027d0b9f3bc2e7fbd282e/sdk/python/feast/ui_server.py#L1122

mlflow_ui_base = (
    mlflow_cfg.get_ui_url() or mlflow.get_tracking_uri() or ""
)

ntkathole · 2026-08-14T05:12:34Z

@Vperiodt please fix linting and conflicts

Signed-off-by: Vanshika Vanshika <vvanshik@redhat.com> rh-pre-commit.version: 2.3.2 rh-pre-commit.check-secrets: ENABLED

Update .secrets.baseline line numbers after MLflow CRD/controller additions so detect-secrets pre-commit passes in CI. Signed-off-by: Vanshika Vanshika <vvanshik@redhat.com> rh-pre-commit.version: 2.3.2 rh-pre-commit.check-secrets: ENABLED

Regenerate .secrets.baseline from master so detect-secrets line numbers match the MLflow operator changes and CI pre-commit stays stable. Signed-off-by: Vanshika Vanshika <vvanshik@redhat.com> rh-pre-commit.version: 2.3.2 rh-pre-commit.check-secrets: ENABLED

Signed-off-by: Vanshika Vanshika <vvanshik@redhat.com> Signed-off-by: Vanshika Vanshika <vvanshik@redhat.com> rh-pre-commit.version: 2.3.2 rh-pre-commit.check-secrets: ENABLED

Use mlflow_cfg.get_ui_url() as the base URL for model links in get_mlflow_feature_models, consistent with get_mlflow_experiments. Falls back to mlflow.get_tracking_uri() when no explicit UI URL is set. Signed-off-by: Vanshika Vanshika <vvanshik@redhat.com>

Vperiodt · 2026-08-14T11:04:10Z

@Vperiodt please fix linting and conflicts

done!

Vperiodt marked this pull request as ready for review July 21, 2026 11:40

Vperiodt requested a review from a team as a code owner July 21, 2026 11:40

ntkathole reviewed Jul 30, 2026

View reviewed changes

Vperiodt force-pushed the feast-mlflow-rhoai branch from 880895c to 7774e4c Compare August 4, 2026 09:11

Vperiodt force-pushed the feast-mlflow-rhoai branch from 81745f4 to b441f95 Compare August 11, 2026 14:46

Vperiodt added 6 commits August 14, 2026 13:43

Feast-mlflow operator support

40409fc

Signed-off-by: Vanshika Vanshika <vvanshik@redhat.com> rh-pre-commit.version: 2.3.2 rh-pre-commit.check-secrets: ENABLED

add auto-discovery

ff60b77

Signed-off-by: Vanshika Vanshika <vvanshik@redhat.com> rh-pre-commit.version: 2.3.2 rh-pre-commit.check-secrets: ENABLED

Auto-discover MLflow UI URL for Feast UI lineage links

4d5478d

Signed-off-by: Vanshika Vanshika <vvanshik@redhat.com> rh-pre-commit.version: 2.3.2 rh-pre-commit.check-secrets: ENABLED

address reviews

ff04395

Signed-off-by: Vanshika Vanshika <vvanshik@redhat.com> Signed-off-by: Vanshika Vanshika <vvanshik@redhat.com> rh-pre-commit.version: 2.3.2 rh-pre-commit.check-secrets: ENABLED

Vperiodt force-pushed the feast-mlflow-rhoai branch from b441f95 to 84b65ed Compare August 14, 2026 08:33

Vperiodt force-pushed the feast-mlflow-rhoai branch from 84b65ed to 098fcbd Compare August 14, 2026 10:57

Navigation Menu

Uh oh!

feat: Add MLflow integration support to Feast operator - #6611

feat: Add MLflow integration support to Feast operator#6611
Vperiodt wants to merge 7 commits into
feast-dev:masterfrom
Vperiodt:feast-mlflow-rhoai

Vperiodt commented Jul 17, 2026 •

edited

Loading

Uh oh!

codecov-commenter commented Jul 17, 2026 •

edited

Loading

Uh oh!

ntkathole Jul 30, 2026

Uh oh!

ntkathole Jul 30, 2026

Uh oh!

ntkathole Jul 30, 2026

Uh oh!

ntkathole Jul 30, 2026

Uh oh!

ntkathole Jul 30, 2026

Uh oh!

ntkathole Jul 30, 2026

Uh oh!

ntkathole commented Aug 11, 2026

Uh oh!

ntkathole commented Aug 14, 2026

Uh oh!

Vperiodt commented Aug 14, 2026

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

3 participants

	if cr.Status.Applied.Mlflow.TrackingUri == nil {
	if cr.Status.Applied.Mlflow != nil && cr.Status.Applied.Mlflow.TrackingUri == nil {

Uh oh!

Conversation

Vperiodt commented Jul 17, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

What this PR does / why we need it:

Which issue(s) this PR fixes:

Checks

Testing Strategy

Misc

Uh oh!

codecov-commenter commented Jul 17, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Codecov Report

Uh oh!

ntkathole Jul 30, 2026

Choose a reason for hiding this comment

Uh oh!

ntkathole Jul 30, 2026

Choose a reason for hiding this comment

Uh oh!

ntkathole Jul 30, 2026

Choose a reason for hiding this comment

Uh oh!

ntkathole Jul 30, 2026

Choose a reason for hiding this comment

Uh oh!

ntkathole Jul 30, 2026

Choose a reason for hiding this comment

Uh oh!

ntkathole Jul 30, 2026

Choose a reason for hiding this comment

Uh oh!

ntkathole commented Aug 11, 2026

Uh oh!

ntkathole commented Aug 14, 2026

Uh oh!

Vperiodt commented Aug 14, 2026

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

3 participants

Vperiodt commented Jul 17, 2026 •

edited

Loading

codecov-commenter commented Jul 17, 2026 •

edited

Loading