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.
This commit is contained in:
vitor-aignosi
2025-10-27 12:28:17 -03:00
parent 22eee9dfba
commit 1cbf356d0c

View File

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