From a5fc526f616d774cbc2d71488827d1fdd778a895 Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Tue, 16 Sep 2025 13:04:43 -0300 Subject: [PATCH] SIENTIAPDE-1222 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. --- laborious/utils/repository/model_repository.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/laborious/utils/repository/model_repository.py b/laborious/utils/repository/model_repository.py index d1021b9..0145554 100644 --- a/laborious/utils/repository/model_repository.py +++ b/laborious/utils/repository/model_repository.py @@ -42,7 +42,7 @@ class MLFlowRepository(): try: model_retention = model_config.get('retention_minutes', 0) 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') transform_keyword = model_config.get( 'transform_function_keyword', 'predict') @@ -79,7 +79,7 @@ class MLFlowRepository(): try: model_retention = model_config.get('retention_minutes', 0) 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') input_index = data.index