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:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user