From a34bdcd8393f8721574c1c295bbc05ec61c45c55 Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Thu, 24 Jul 2025 11:24:21 -0300 Subject: [PATCH] SIENTIAPDE-1171 Refactor data merging in MLFlowRepository to simplify merge operation in fit method. Removed unnecessary parameters for improved readability and maintainability. --- laborious/utils/repository/model_repository.py | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/laborious/utils/repository/model_repository.py b/laborious/utils/repository/model_repository.py index c3ef12d..2f10a9d 100644 --- a/laborious/utils/repository/model_repository.py +++ b/laborious/utils/repository/model_repository.py @@ -135,13 +135,7 @@ class MLFlowRepository(): target_name = data_model.target_variable y = data[target_name] treated_data = pd.merge( - 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 - ) + treated_data, y, left_index=True, right_index=True) prediction_model = prediction_model.fit(treated_data) experiment = self.get_experiment_by_run_id(latest_production_id) mlflow.set_experiment(experiment)