If you don't need the epoch_raw I see 2 options : don't return it (say it with a parameter of _get_epoch_from_disk) OR have epoch_raw be the same as epoch (less clear). We can then avoir this self._preprocess_epoch(epoch_raw, 'reload'). I find the "reload" confusing.

sorry for being so picky...