From c75a5921f317af690d26dd8bc9fc880299e2887e Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Thu, 16 Apr 2026 09:34:36 -0300 Subject: [PATCH] feat: add run_name to TrainModelResult for enhanced tracking - Introduced run_name parameter in TrainModelResult to improve identification of training runs based on model type, name, and experiment run ID. --- model_manager/utils/repository/data_manager_repository.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/model_manager/utils/repository/data_manager_repository.py b/model_manager/utils/repository/data_manager_repository.py index 0406761..73c46bc 100644 --- a/model_manager/utils/repository/data_manager_repository.py +++ b/model_manager/utils/repository/data_manager_repository.py @@ -200,7 +200,9 @@ class DataManagerRepository(SientiaMonitoring): metadata, ) - return TrainModelResult(params=params, train_data=train_data, val_data=val_data) + run_name = f'train_model_{params.model_type}_{params.model_name}_{params.experiment_run_id}' + + return TrainModelResult(params=params, train_data=train_data, val_data=val_data, run_name=run_name) def _as_series(self, pred: pd.DataFrame | pd.Series) -> pd.Series: if isinstance(pred, pd.Series):