Merge pull request columnflow#426 from columnflow/fix/btag_weight_cferr

multiply cferr columns with nominal btag weight
haddadanas · May 8, 2024 · f8a1606 · f8a1606
2 parents 39e031e + e01e357
commit f8a1606
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/columnflow/production/cms/btag.py b/columnflow/production/cms/btag.py
@@ -191,6 +191,14 @@ def add_weight(syst_name, syst_direction, column_name):
                     direction,
                     f"btag_weight_{name}_{direction}",
                 )
+                if syst_name in ["cferr1", "cferr2"]:
+                    # for c flavor uncertainties, multiply the uncertainty with the nominal btag weight
+                    events = set_ak_column(
+                        events,
+                        f"btag_weight_{name}_{direction}",
+                        events.btag_weight * events[f"btag_weight_{name}_{direction}"],
+                        value_type=np.float32,
+                    )
     elif self.shift_is_known_jec_source:
         # TODO: year dependent jec variations fully covered?
         events = add_weight(