Fix small bug in CNF (#1479)

XENONnT · Nov 16, 2024 · e222b25 · e222b25
1 parent ded7eab
commit e222b25
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/straxen/plugins/peaks/peak_positions_cnf.py b/straxen/plugins/peaks/peak_positions_cnf.py
@@ -237,7 +237,7 @@ def compute(self, peaks):
 
         theta_array = np.arctan2(contours[..., 1], contours[..., 0])
 
-        avg_theta = np.arctan2(result[f"y_{self.algorithm}"], result[f"x_{self.algorithm}"])
+        avg_theta = np.arctan2(xy[:, 1], xy[:, 0])
 
         theta_diff = self.calculate_theta_diff(theta_array, avg_theta)