FIX: sample_epochs feature