diff --git a/pipt/misc_tools/extract_tools.py b/pipt/misc_tools/extract_tools.py index 622705d7..c37ac91e 100644 --- a/pipt/misc_tools/extract_tools.py +++ b/pipt/misc_tools/extract_tools.py @@ -296,6 +296,7 @@ def organize_sparse_representation(info: Union[dict,list]) -> dict: sparse['keep_ca'] = info.get('keep_ca', False) sparse['inactive_value'] = info['inactive_value'] sparse['use_ensemble'] = info.get('use_ensemble', None) + if sparse['use_ensemble'] == False: sparse['use_ensemble'] = None return sparse