Fix feature selection in `predict_marginals` unexpectedly erasing data

3 jobs for fix-fet in 15 minutes and 33 seconds (queued for 2 seconds)