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):