From 19c8a028d23688881972cf82ac19f3a128e24e45 Mon Sep 17 00:00:00 2001
From: vitor-aignosi <vitor.santos@aignosi.com.br>
Date: Thu, 11 Jun 2026 10:02:11 -0300
Subject: [PATCH] SIENTIAPDE-1646  SIENTIAPDE-1646 Update prediction data
 handling in MLFlow

- Renamed the prediction column in the MLFlow retraining process to improve clarity and consistency in the output data.
---
 laborious/activities/mlflow.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/laborious/activities/mlflow.py b/laborious/activities/mlflow.py
index ba6a346..3ac5d95 100644
--- a/laborious/activities/mlflow.py
+++ b/laborious/activities/mlflow.py
@@ -529,6 +529,7 @@ class MLFlow(SientiaMonitoring):
 
             # Keep heavy model fitting outside MLflow run timing.
             prediction_data = wrapper.retrain(data)
+            prediction_data.rename(columns={target: 'prediction'}, inplace=True)
 
             # Merge prediction data with retrain data
             evaluation_data = pd.merge(