Skip to content

Commit f8135e5

Browse files
committed
Add BaseSpectrum support to equalize_channels
1 parent 5829fa0 commit f8135e5

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

mne/channels/channels.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -148,21 +148,24 @@ def equalize_channels(instances, copy=True, verbose=None):
148148
from ..forward import Forward
149149
from ..io import BaseRaw
150150
from ..time_frequency import BaseTFR, CrossSpectralDensity
151+
from ..time_frequency.spectrum import BaseSpectrum
151152

152153
# Instances need to have a `ch_names` attribute and a `pick_channels`
153154
# method that supports `ordered=True`.
154155
allowed_types = (
155156
BaseRaw,
156157
BaseEpochs,
157158
Evoked,
159+
BaseSpectrum,
158160
BaseTFR,
159161
Forward,
160162
Covariance,
161163
CrossSpectralDensity,
162164
Info,
163165
)
164166
allowed_types_str = (
165-
"Raw, Epochs, Evoked, TFR, Forward, Covariance, CrossSpectralDensity or Info"
167+
"Raw, Epochs, Evoked, Spectrum, TFR, Forward, Covariance, CrossSpectralDensity "
168+
"or Info"
166169
)
167170
for inst in instances:
168171
_validate_type(

0 commit comments

Comments
 (0)