Debug

XENONnT · Nov 20, 2024 · 7db69bf · 7db69bf
1 parent eee7105
commit 7db69bf
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/straxen/plugins/events/event_position_uncertainty.py b/straxen/plugins/events/event_position_uncertainty.py
@@ -2,7 +2,7 @@
 import strax
 import straxen
 from straxen.plugins.defaults import DEFAULT_POSREC_ALGO
-from straxen.plugins.peaks.peak_positions_cnf import PeakPositionsCNF
+from straxen.plugins.peaklets.peaklet_positions_cnf import PeakletPositionsCNF
 
 
 export, __all__ = strax.exporter()
@@ -305,7 +305,7 @@ def compute(self, events):
 
             avg_theta = np.arctan2(events[f"{type_}_y_cnf"], events[f"{type_}_x_cnf"])
 
-            theta_diff = PeakPositionsCNF.calculate_theta_diff(theta_array, avg_theta)
+            theta_diff = PeakletPositionsCNF.calculate_theta_diff(theta_array, avg_theta)
 
             # Store uncertainties
             result[f"{type_}_r_position_uncertainty"] = (r_max - r_min) / 2

diff --git a/tests/plugins/posrec_plugins.py b/tests/plugins/posrec_plugins.py
@@ -73,7 +73,7 @@ def test_posrec_bad_configs_raising_errors(
     with self.assertRaises(FileNotFoundError):
         plugin.get_tf_model()
 
-    dummy_st.register(straxen.plugins._peaklet_positions_base.PeakletPositionsBase)
+    dummy_st.register(straxen.plugins.peaklets._peaklet_positions_base.PeakletPositionsBase)
     plugin_name = strax.camel_to_snake("PeakletPositionsBase")
     with self.assertRaises(NotImplementedError):
         dummy_st.get_single_plugin(self.run_id, plugin_name)

diff --git a/tests/plugins/s1_posrec_plugins.py b/tests/plugins/s1_posrec_plugins.py
@@ -76,7 +76,7 @@ def test_posrec_bad_configs_raising_errors(
     with self.assertRaises(FileNotFoundError):
         plugin.get_tf_model()
 
-    dummy_st.register(straxen.plugins._peaklet_positions_base.PeakletPositionsBase)
+    dummy_st.register(straxen.plugins.peaklets._peaklet_positions_base.PeakletPositionsBase)
     plugin_name = strax.camel_to_snake("PeakletPositionsBase")
     with self.assertRaises(NotImplementedError):
         dummy_st.get_single_plugin(self.run_id, plugin_name)