Merge branch 'master' into fix/qcd_with_systs

uhh-cms · Jan 22, 2025 · 47895d7 · 47895d7
2 parents 2d45d2d + 71b3dbf
commit 47895d7
Show file tree

Hide file tree

Showing 2 changed files with 409 additions and 61 deletions.
diff --git a/hbt/config/configs_hbt.py b/hbt/config/configs_hbt.py
@@ -524,7 +524,19 @@ def if_not_era(*, values: list[str | None] | None = None, **kwargs) -> list[str]
 
     # variable groups for conveniently looping over certain variables
     # (used during plotting)
-    cfg.x.variable_groups = {}
+    cfg.x.variable_groups = {
+        "hh": (hh := [f"hh_{var}" for var in ["energy", "mass", "pt", "eta", "phi", "dr"]]),
+        "dilep": (dilep := [f"dilep_{var}" for var in ["energy", "mass", "pt", "eta", "phi", "dr"]]),
+        "dijet": (dijet := [f"dijet_{var}" for var in ["energy", "mass", "pt", "eta", "phi", "dr"]]),
+        "default": [
+            *dijet,
+            *dilep,
+            *hh,
+            "mu1_pt", "mu1_eta", "mu1_phi", "mu2_pt", "mu2_eta", "mu2_phi",
+            "e1_pt", "e1_eta", "e1_phi", "e2_pt", "e2_eta", "e2_phi",
+            "tau1_pt", "tau1_eta", "tau1_phi", "tau2_pt", "tau2_eta", "tau2_phi",
+        ],
+    }
 
     # shift groups for conveniently looping over certain shifts
     # (used during plotting)