legend-dataflow-scripts 0.1.3__tar.gz → 0.1.4__tar.gz

{legend_dataflow_scripts-0.1.3/src/legend_dataflow_scripts.egg-info → legend_dataflow_scripts-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: legend-dataflow-scripts
-Version: 0.1.3
+Version: 0.1.4
 Summary: Python package for the processing scripts for LEGEND-200 data
 Author-email: George Marshall <ggmarsh@uw.edu>, Luigi Pertoldi <gipert@pm.me>
 Maintainer: The LEGEND Collaboration

{legend_dataflow_scripts-0.1.3 → legend_dataflow_scripts-0.1.4/src/legend_dataflow_scripts.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: legend-dataflow-scripts
-Version: 0.1.3
+Version: 0.1.4
 Summary: Python package for the processing scripts for LEGEND-200 data
 Author-email: George Marshall <ggmarsh@uw.edu>, Luigi Pertoldi <gipert@pm.me>
 Maintainer: The LEGEND Collaboration

{legend_dataflow_scripts-0.1.3 → legend_dataflow_scripts-0.1.4}/src/legenddataflowscripts/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.1.3'
-__version_tuple__ = version_tuple = (0, 1, 3)
+__version__ = version = '0.1.4'
+__version_tuple__ = version_tuple = (0, 1, 4)

{legend_dataflow_scripts-0.1.3 → legend_dataflow_scripts-0.1.4}/src/legenddataflowscripts/par/geds/hit/qc.py

@@ -129,6 +129,7 @@ def build_qc(
     else:
         hit_dict_fft = {}
         plot_dict_fft = {}
+        fft_data = None
 
     if overwrite is not None:
         for name in kwarg_dict_fft["cut_parameters"]:
@@ -234,7 +235,7 @@ def build_qc(
         exp = info["expression"]
         for key in info.get("parameters", None):
             exp = re.sub(f"(?<![a-zA-Z0-9]){key}(?![a-zA-Z0-9])", f"@{key}", exp)
-        if outname not in fft_data:
+        if fft_data is not None and outname not in fft_data:
             fft_data[outname] = fft_data.eval(
                 exp, local_dict=info.get("parameters", None)
             )
@@ -251,24 +252,34 @@ def build_qc(
                 ((sf_cal) * (1 - sf_cal))
                 / len(data.query("~is_pulser & ~is_recovering"))
             )
-            sf_fft = len(fft_data.query(f"{entry} & ~is_recovering")) / len(
-                fft_data.query("~is_recovering")
-            )
-            sf_fft_err = 100 * np.sqrt(
-                ((sf_fft) * (1 - sf_fft)) / len(fft_data.query("~is_recovering"))
-            )
             sf_cal *= 100
-            sf_fft *= 100
-
-            msg = f"{entry} cut applied: {sf_cal:.2f}% of events passed the cut for cal data, {sf_fft:.2f}% for fft data"
+            msg = f"{entry} cut applied: {sf_cal:.2f}% of events passed the cut for cal data"
             log.info(msg)
+
             qc_results[entry] = {
                 "sf_cal": sf_cal,
                 "sf_cal_err": sf_cal_err,
-                "sf_fft": sf_fft,
-                "sf_fft_err": sf_fft_err,
             }
 
+            if fft_data is not None:
+                sf_fft = len(fft_data.query(f"{entry} & ~is_recovering")) / len(
+                    fft_data.query("~is_recovering")
+                )
+                sf_fft_err = 100 * np.sqrt(
+                    ((sf_fft) * (1 - sf_fft)) / len(fft_data.query("~is_recovering"))
+                )
+                sf_fft *= 100
+                msg = f"{entry} cut applied: {sf_fft:.2f}% of events passed the cut for fft data"
+
+                log.info(msg)
+
+                qc_results[entry].update(
+                    {
+                        "sf_fft": sf_fft,
+                        "sf_fft_err": sf_fft_err,
+                    }
+                )
+
     out_dict = convert_dict_np_to_float(
         {"operations": hit_dict, "results": {"qc": qc_results}}
     )

{legend_dataflow_scripts-0.1.3 → legend_dataflow_scripts-0.1.4}/src/legenddataflowscripts/workflow/__init__.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 from .execenv import execenv_prefix, execenv_pyexe
+from .pre_compile_catalog import pre_compile_catalog
 from .utils import (
     as_ro,
     set_last_rule_name,
@@ -13,6 +14,7 @@ __all__ = [
     "as_ro",
     "execenv_prefix",
     "execenv_pyexe",
+    "pre_compile_catalog",
     "set_last_rule_name",
     "subst_vars",
     "subst_vars_impl",