diff --git a/src/pipt/misc_tools/extract_tools.py b/src/pipt/misc_tools/extract_tools.py index e62a0f9..6dd0fd5 100644 --- a/src/pipt/misc_tools/extract_tools.py +++ b/src/pipt/misc_tools/extract_tools.py @@ -443,8 +443,11 @@ def organize_sparse_representation(info: Union[dict,list]) -> dict: mask = np.load(filename)['mask'] sparse['mask'].append(mask.flatten()) + if 'compress_data' not in info: + raise KeyError("Missing required key: 'compress_data' specifying data to be compressed") + # Read rest of keywords - sparse['compress_data'] = info.get('compress_data', False) + sparse['compress_data'] = info['compress_data'] sparse['level'] = info['level'] sparse['wname'] = info['wname'] sparse['threshold_rule'] = info['threshold_rule']