removed particle_id and renamed kinematic_id

uhh-cms · Feb 12, 2025 · d05670c · d05670c
1 parent 423c2a2
commit d05670c
Showing 1 changed file with 7 additions and 16 deletions.
diff --git a/hbt/tasks/sync.py b/hbt/tasks/sync.py
@@ -6,7 +6,7 @@
 
 from __future__ import annotations
 
-from functools import reduce, partial
+from functools import reduce
 from operator import or_
 
 import luigi
@@ -318,24 +318,16 @@ def get_mapping(config_inst, demanded_categories):
             events = select_leptons(events, {"rawDeepTau2018v2p5VSjet": empty[np.float32][1]})
 
             # find ids for specific categories
-            cat_mapper = partial(
-                get_category_id,
+            category_id = get_category_id(
                 category_id=events.category_ids,
                 config_inst=self.config_inst,
+                category_replacement_map={
+                    "res1b": 0,
+                    "res2b": 1,
+                    "boosted": 2},
                 axis=-1,
             )
 
-            kinematic_id = cat_mapper(category_replacement_map={
-                "res1b": 0,
-                "res2b": 1,
-                "boosted": 2},
-            )
-
-            particle_enriched_id = cat_mapper(category_replacement_map={
-                "dy": 0,
-                "tt": 1},
-            )
-
             # project into dataframe
             met_name = self.config_inst.x.met_name
             df = ak.to_dataframe({
@@ -345,8 +337,7 @@ def get_mapping(config_inst, demanded_categories):
                 "lumi": events.luminosityBlock,
                 # high-level events variables
                 "channel_id": events.channel_id,
-                "kinematic_id": kinematic_id,
-                "particle_enriched_id": particle_enriched_id,
+                "category_id": category_id,
                 "os": events.leptons_os * 1,
                 "iso": events.tau2_isolated * 1,
                 "deterministic_seed": uint64_to_str(events.deterministic_seed),