From a5d2b0d3fda9425584bc44a5571df41a38957164 Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Mon, 13 Oct 2025 15:59:10 -0300 Subject: [PATCH] SIENTIAPDE-1231 Refactor `force_memory_release` function in model_repository.py to improve logging - Updated the `force_memory_release` function to remove metadata parameter and enhance logging by using `logger.info()` instead of `logger.custom_info()`. - Adjusted calls to `force_memory_release` in the MLFlowRepository to pass the logger instance, ensuring consistent logging during memory management operations. --- laborious/utils/repository/model_repository.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/laborious/utils/repository/model_repository.py b/laborious/utils/repository/model_repository.py index 25345ce..691857a 100644 --- a/laborious/utils/repository/model_repository.py +++ b/laborious/utils/repository/model_repository.py @@ -36,16 +36,16 @@ TRANSFORMED_COMPRESSED_PATH = "artifacts/training_transformer.pkl" PREDICTION_COMPRESSED_PATH = "artifacts/stacking_model.pkl" -def force_memory_release(logger: Logger, metadata: dict): +def force_memory_release(logger: Logger): gc.collect() try: ctypes.CDLL("libc.so.6").malloc_trim(0) - logger.custom_info( - f"Memory released", metadata) + logger.info( + f"Memory released") except Exception as e: - logger.custom_info( - f"Memory release failed: {e}", metadata) + logger.info( + f"Memory release failed: {e}") class MLFlowRepository(): @@ -630,7 +630,7 @@ class MLFlowRepository(): f"Deleting model {model_name} from memory") del model - force_memory_release() + force_memory_release(self.logger) return prediction @@ -659,7 +659,7 @@ class MLFlowRepository(): f"Deleting model {model_name} from memory") del model - force_memory_release() + force_memory_release(self.logger) return prediction