From ca18f9d4ce9745bb6164d7e231dc93f4aad901a6 Mon Sep 17 00:00:00 2001 From: timonmerk Date: Tue, 21 Nov 2023 16:23:11 +0100 Subject: [PATCH] Update nm_bispectra.py --- py_neuromodulation/nm_bispectra.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/py_neuromodulation/nm_bispectra.py b/py_neuromodulation/nm_bispectra.py index 86bde01d..522be503 100644 --- a/py_neuromodulation/nm_bispectra.py +++ b/py_neuromodulation/nm_bispectra.py @@ -99,12 +99,12 @@ def calc_feature(self, data: np.array, features_compute: dict) -> dict: waveshape = WaveShape( data=fft_coeffs, - freqs=freqs.astype(int), + freqs=freqs, sampling_freq=self.sfreq, verbose=False, ) - waveshape.compute(f1s=tuple(self.f1s), f2s=tuple(self.f2s)) + waveshape.compute(f1s=tuple(self.f1s[0], self.f1s[-1]), f2s=tuple(self.f2s[0], self.f2s[-1])) bispectrum = np.squeeze(waveshape.results._data)