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 key for compression in MLFlowRepository

- Changed the key for compression from 'compressed' to 'is_compressed' in model configuration handling to ensure consistency with updated standards.
This commit is contained in:
vitor-aignosi
2025-09-16 13:04:43 -03:00
parent 2b45c1cbd4
commit a5fc526f61
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

@@ -42,7 +42,7 @@ class MLFlowRepository():
try: try:
model_retention = model_config.get('retention_minutes', 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('compressed', False) compressed = model_config.get('is_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')
@@ -79,7 +79,7 @@ class MLFlowRepository():
try: try:
model_retention = model_config.get('retention_minutes', 0) model_retention = model_config.get('retention_minutes', 0)
flavor = model_config.get('predict_flavor', 'pyfunc') flavor = model_config.get('predict_flavor', 'pyfunc')
compressed = model_config.get('compressed', False) compressed = model_config.get('is_compressed', False)
retention_target = model_config.get('retention_target', 'model') retention_target = model_config.get('retention_target', 'model')
input_index = data.index input_index = data.index