gitea_admin/sientia-dataops-laborious_temporal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SIENTIAPDE-1314

Browse Source 
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
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
laborious/utils/repository/model_repository.py
View File

@@ -441,7 +441,7 @@ class MLFlowRepository:
""" """
self.logger.debug(f'Model {model_name} is outdated, downloading a new one') 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] del self.model_cache[model_key]
def get_model(self, model_name: str, retention: int, model_type: str, flavor: str) -> Any: def get_model(self, model_name: str, retention: int, model_type: str, flavor: str) -> Any: