update slope calculation

splendor-collab · Nov 8, 2024 · 4b9960d · 4b9960d
1 parent 9c5bdd5
commit 4b9960d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/splendsp/dsp/_utils.py b/src/splendsp/dsp/_utils.py
@@ -116,8 +116,9 @@ def slope(x, y, removemeans=True):
     """
 
     x_mean = np.mean(x) if removemeans else 0
+    y_mean = np.mean(y) if removemeans else 0
 
-    return np.sum((x - x_mean) * (y - x_mean)) / np.sum((x - x_mean) ** 2)
+    return np.sum((x - x_mean) * (y - y_mean)) / np.sum((x - x_mean) ** 2)
 
 
 def lowpassfilter(traces, cut_off_freq=100000, fs=625e3, order=1):