From e6c55721847478ae55fbdf90b4267271d054b6a4 Mon Sep 17 00:00:00 2001 From: Bogdan Wiederspan Date: Thu, 19 Dec 2024 13:42:12 +0100 Subject: [PATCH] prevent throwout of trigger coulmns for empty selector --- hbt/selection/default.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/hbt/selection/default.py b/hbt/selection/default.py index 9390b3c1..6944a15a 100644 --- a/hbt/selection/default.py +++ b/hbt/selection/default.py @@ -209,7 +209,7 @@ def empty_init(self: Selector) -> None: # add custom columns self.uses.add("Jet.phi") # needed by vector behavior for accessing pt in btag_weights - self.produces |= {"channel_id", "leptons_os", "tau2_isolated"} + self.produces |= {"channel_id", "leptons_os", "tau2_isolated", "{single,cross}_triggered"} @empty.call @@ -273,6 +273,8 @@ def empty_call( events = set_ak_column(events, "channel_id", np.zeros(len(events), dtype=np.uint8)) events = set_ak_column(events, "leptons_os", np.zeros(len(events), dtype=bool)) events = set_ak_column(events, "tau2_isolated", np.zeros(len(events), dtype=bool)) + events = set_ak_column(events, "cross_triggered", np.zeros(len(events), dtype=bool)) + events = set_ak_column(events, "single_triggered", np.zeros(len(events), dtype=bool)) # trivial selection mask capturing all events results.event = np.ones(len(events), dtype=bool)