Debug

XENONnT · Jan 24, 2025 · 44cd96b · 44cd96b
1 parent 9806c55
commit 44cd96b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/straxen/config/protocols.py b/straxen/config/protocols.py
@@ -97,6 +97,7 @@ def open_neural_net(model_path: str, custom_objects=None, **kwargs):
     """Open a tensorflow file and return a keras model."""
     # Nested import to reduce loading time of import straxen and it not
     # base requirement
+    import jax  # noqa: F401
     import tensorflow as tf
 
     if not os.path.exists(model_path):

diff --git a/tests/plugins/peak_building.py b/tests/plugins/peak_building.py
@@ -68,7 +68,7 @@ def test_tight_coincidence(self: PluginTestCase):
     peaklets = self.st.get_array(self.run_id, "peaklets", progress_bar=False)
     message = "There might be some issue in tight_coincidence."
     sum_tight_coincidence = np.sum(peaklets["tight_coincidence"])
-    assert np.abs(sum_tight_coincidence - 1990) < 5, message
+    assert sum_tight_coincidence == 1991, message
 
 
 if __name__ == "__main__":