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

SIENTIAPDE-1222

Browse Source 
Update model configuration keys in MLFlowRepository for consistency

- Changed 'model_retention' to 'retention_minutes' and 'is_compressed' to 'compressed' in model configuration handling, ensuring alignment with updated configuration standards.
This commit is contained in:
vitor-aignosi
2025-09-16 12:34:56 -03:00
parent b1951ea2c0
commit 2b45c1cbd4
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -40,9 +40,9 @@ class MLFlowRepository():
""" """
try: try:
model_retention = model_config.get('model_retention', 0) model_retention = model_config.get('retention_minutes', 0)
flavor = model_config.get('transform_flavor', 'sklearn') flavor = model_config.get('transform_flavor', 'sklearn')
compressed = model_config.get('is_compressed', False) compressed = model_config.get('compressed', False)
retention_target = model_config.get('retention_target', 'model') retention_target = model_config.get('retention_target', 'model')
transform_keyword = model_config.get( transform_keyword = model_config.get(
'transform_function_keyword', 'predict') 'transform_function_keyword', 'predict')