it should work indeed if epochs is a 3d array or instance of Epochs. Should be updated by @trachelr