update resolution of eec in lepton selection

uhh-cms · Feb 3, 2025 · 67f52a0 · 67f52a0
1 parent 7574423
commit 67f52a0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/hbt/selection/lepton.py b/hbt/selection/lepton.py
@@ -160,8 +160,10 @@ def electron_selection(
 def electron_selection_init(self) -> None:
     from columnflow.config_util import get_shifts_from_sources
     if self.config_inst.campaign.x.run == 3 and self.config_inst.campaign.x.year == 2022:
-        self.shifts.update(get_shifts_from_sources(self.config_inst, "eec"))
-        self.shifts.update(get_shifts_from_sources(self.config_inst, "eer"))
+        self.shifts |= {
+            shift_inst.name for shift_inst in self.config_inst.shifts
+                if shift_inst.has_tag(("ees", "eer"))
+        }
 
 
 @selector(