From 1cbf356d0c5d83b99d53cf9370f01cfe903f2805 Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Mon, 27 Oct 2025 12:28:17 -0300 Subject: [PATCH] SIENTIAPDE-1314 Refactor model cache deletion in MLFlowRepository - Updated the model cache deletion logic to remove the entire 'target' key instead of just the 'model' subkey, streamlining the cache management process. --- laborious/utils/repository/model_repository.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/laborious/utils/repository/model_repository.py b/laborious/utils/repository/model_repository.py index 7400502..4572f09 100644 --- a/laborious/utils/repository/model_repository.py +++ b/laborious/utils/repository/model_repository.py @@ -441,7 +441,7 @@ class MLFlowRepository: """ self.logger.debug(f'Model {model_name} is outdated, downloading a new one') - del self.model_cache[model_key]['target']['model'] + del self.model_cache[model_key]['target'] del self.model_cache[model_key] def get_model(self, model_name: str, retention: int, model_type: str, flavor: str) -> Any: