diff --git a/laborious/utils/repository/model_repository.py b/laborious/utils/repository/model_repository.py index 18b4620..d1021b9 100644 --- a/laborious/utils/repository/model_repository.py +++ b/laborious/utils/repository/model_repository.py @@ -40,9 +40,9 @@ class MLFlowRepository(): """ try: - model_retention = model_config.get('model_retention', 0) + model_retention = model_config.get('retention_minutes', 0) 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') transform_keyword = model_config.get( 'transform_function_keyword', 'predict')