From 1261e4a8aab7990bc63337b9a4a23c32e519c8d0 Mon Sep 17 00:00:00 2001
From: vitor-aignosi <vitor.santos@aignosi.com.br>
Date: Thu, 24 Jul 2025 10:56:05 -0300
Subject: [PATCH] SIENTIAPDE-1171  Enhance data merging in MLFlowRepository by
 specifying the 'on' parameter for improved join accuracy. This change ensures
 that the merge operation correctly aligns data based on the 'id' field.

---
 laborious/utils/repository/model_repository.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/laborious/utils/repository/model_repository.py b/laborious/utils/repository/model_repository.py
index 888b5c8..c3ef12d 100644
--- a/laborious/utils/repository/model_repository.py
+++ b/laborious/utils/repository/model_repository.py
@@ -138,6 +138,7 @@ class MLFlowRepository():
             treated_data, y,
             left_index=True,
             right_index=True,
+            on='id',
             how='left',           # or 'inner', depending on your needs
             validate='one_to_one'  # ensures each index appears only once in both
         )