From b245801e094df9b2d7894e173cb53af8c475df80 Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Thu, 16 Apr 2026 15:08:26 -0300 Subject: [PATCH] fix: clear input example in Training class before model storage - Set wrapper._input_example to None to ensure no residual data is stored with the model, enhancing data integrity during model storage. --- model_manager/activities/training.py | 1 + 1 file changed, 1 insertion(+) diff --git a/model_manager/activities/training.py b/model_manager/activities/training.py index fb14f98..ae9bae0 100644 --- a/model_manager/activities/training.py +++ b/model_manager/activities/training.py @@ -338,6 +338,7 @@ class Training(SientiaMonitoring): raise ValueError('Report path, train data path, or test data path is not set') self.info(f'Storing model for {train_params.model_type}', metadata) + wrapper._input_example = None wrapper.store_model(name=train_params.model_name) self.info(f'Logging artifacts for {train_params.model_type}', metadata) mlflow.log_artifact(train_result.report_path)