Use column collections.

uhh-cms · Jan 5, 2024 · 116d7df · 116d7df
1 parent 0c52163
commit 116d7df
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/.gitignore b/.gitignore
@@ -35,3 +35,5 @@ data
 .data
 .law
 .setups
+.mypy_cache
+.vscode
diff --git a/hbt/config/configs_run2ul.py b/hbt/config/configs_run2ul.py
@@ -20,6 +20,7 @@
     get_root_processes_from_campaign, add_shift_aliases, get_shifts_from_sources,
     verify_config_processes,
 )
+from columnflow.columnar_util import ColumnCollection
 
 
 thisdir = os.path.dirname(os.path.abspath(__file__))
@@ -661,9 +662,7 @@ def add_config(
             "MET.pt", "MET.phi", "MET.significance", "MET.covXX", "MET.covXY", "MET.covYY",
             "PV.npvs",
             # columns added during selection
-            "channel_id", "process_id", "category_ids", "mc_weight", "pdf_weight*", "murmuf_weight*",
-            "leptons_os", "tau2_isolated", "single_triggered", "cross_triggered",
-            "deterministic_seed", "pu_weight*", "btag_weight*", "cutflow.*",
+            ColumnCollection.ALL_FROM_SELECTOR,
         },
         "cf.MergeSelectionMasks": {
             "cutflow.*",

diff --git a/modules/columnflow b/modules/columnflow