Skip to content

Commit f65c6ff

Browse files
committed
Add BaseSpectrum support to equalize_channels
1 parent 9970507 commit f65c6ff

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

mne/channels/channels.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -120,8 +120,11 @@ def equalize_channels(instances, copy=True, verbose=None):
120120
----------
121121
instances : list
122122
A list of MNE-Python objects to equalize the channels for. Objects can
123-
be of type Raw, Epochs, Evoked, AverageTFR, Forward, Covariance,
123+
be of type Raw, Epochs, Evoked, Spectrum, AverageTFR, Forward, Covariance,
124124
CrossSpectralDensity or Info.
125+
126+
.. versionchanged:: 1.11
127+
Added support for :class:`mne.time_frequency.Spectrum` objects.
125128
copy : bool
126129
When dropping and/or re-ordering channels, an object will be copied
127130
when this parameter is set to ``True``. When set to ``False`` (the
@@ -148,21 +151,24 @@ def equalize_channels(instances, copy=True, verbose=None):
148151
from ..forward import Forward
149152
from ..io import BaseRaw
150153
from ..time_frequency import BaseTFR, CrossSpectralDensity
154+
from ..time_frequency.spectrum import BaseSpectrum
151155

152156
# Instances need to have a `ch_names` attribute and a `pick_channels`
153157
# method that supports `ordered=True`.
154158
allowed_types = (
155159
BaseRaw,
156160
BaseEpochs,
157161
Evoked,
162+
BaseSpectrum,
158163
BaseTFR,
159164
Forward,
160165
Covariance,
161166
CrossSpectralDensity,
162167
Info,
163168
)
164169
allowed_types_str = (
165-
"Raw, Epochs, Evoked, TFR, Forward, Covariance, CrossSpectralDensity or Info"
170+
"Raw, Epochs, Evoked, Spectrum, TFR, Forward, Covariance, CrossSpectralDensity "
171+
"or Info"
166172
)
167173
for inst in instances:
168174
_validate_type(

0 commit comments

Comments
 (0)