added masked data to channel specific rejection

Author: CarinaFo

mne/epochs.py

@@ -4675,4 +4675,6 @@ def channel_specific_epoch_rejection(data: np.ndarray,
     mean = np.mean(abs_data, axis=(-1, 0))  # (channels X (frequency))
     # keep epochs where the maximum voltage is smaller than the mean + (outliers * std)
     keep = max < ((outliers * std) + mean)  # (epochs X channels X (frequency))
-    return keep
+    # set values in 3D data arry to NaN where 2D mask is False
+    data[keep is False] = np.nan
+    return keep, data