Python package for sleep staging from EEG data