Add additional sparse probing datasets

[adamkarvonen]

A few changes here:

• Adding more sparse probing datasets, cleaning up dataset creation code
• Speeding up activation collection
• Add retries due to intermittent dataset loading errors in unlearning
• Add a single shift scr metric key