fix multiple bsqi

aim-lab · Mar 29, 2023 · a729eb5 · a729eb5
1 parent b90e45d
commit a729eb5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pecg/Preprocessing.py b/pecg/Preprocessing.py
@@ -149,7 +149,7 @@ def bsqi(self, peaks: np.array = np.array([]), test_peaks: np.array = np.array([
                 else:
                     testqrs = test_peaks
 
-                bsqi[i] = self.__calculate_bsqi(refqrs[refqrs[:, i] > 0, i], testqrs[testqrs[:, i] > 0, i], fs)
+                bsqi[i] = self.__calculate_bsqi(refqrs[refqrs > 0], testqrs[testqrs > 0], fs)
         elif len(np.shape(signal)) == 1:
             fp = FiducialPoints(signal, fs)
             if not peaks.any():