Update nm_bispectra.py

neuromodulation · Nov 21, 2023 · ca18f9d · ca18f9d
1 parent 7249bfb
commit ca18f9d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)