Skip to content

Commit b6de612

Browse files
committed
tweak printing in event object
1 parent e5161d0 commit b6de612

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

specparam/objs/event.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -229,6 +229,11 @@ def fit(self, freqs=None, spectrograms=None, freq_range=None, peak_org=None,
229229
if spectrograms is not None:
230230
self.add_data(freqs, spectrograms, freq_range)
231231

232+
# If 'verbose', print out a marker of what is being run
233+
if self.verbose and not progress:
234+
print('Fitting model across {} events of {} windows.'.format(\
235+
len(self.spectrograms), self.n_time_windows))
236+
232237
if n_jobs == 1:
233238
self._reset_event_results(len(self.spectrograms))
234239
for ind, spectrogram in _progress(enumerate(self.spectrograms), progress, len(self)):
@@ -546,6 +551,17 @@ def convert_results(self, peak_org):
546551
self.event_time_results = event_group_to_dict(self.event_group_results, peak_org)
547552

548553

554+
def _check_width_limits(self):
555+
"""Check and warn about bandwidth limits / frequency resolution interaction."""
556+
557+
# Only check & warn on first spectrogram
558+
# This is to avoid spamming standard output for every spectrogram in the set
559+
if np.all(self.spectrograms[0] == self.spectrogram):
560+
#if self.power_spectra[0, 0] == self.power_spectrum[0]:
561+
super()._check_width_limits()
562+
563+
564+
549565
def _par_fit(spectrogram, model):
550566
"""Helper function for running in parallel."""
551567

0 commit comments

Comments
 (0)