don't import sklearn here otherwise you make it a strong dependency. People should be apply to use mne wiithout installing sklearn