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)