Still get NaNs after defining the noise_csd from all epochs (N=2) and selecting a single epoch afterwards.