Nitpick: `add_raw_scores` should just be `raw_scores`, since they are the `raw_scores` that are appended when we `keep_raw_scores`.