added only inclusive weights

haddadanas · Oct 28, 2024 · 1dc03e1 · 1dc03e1
1 parent 31cfc08
commit 1dc03e1
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 1 deletion.
diff --git a/hbt/config/configs_hbt.py b/hbt/config/configs_hbt.py
@@ -104,7 +104,6 @@ def if_era(
     dy_group = [
         "dy_m4to10",
         "dy_m10to50",
-        # "dy_m50toinf",
         "dy_m50toinf_0j",
         "dy_m50toinf_1j_pt0to40", "dy_m50toinf_1j_pt40to100", "dy_m50toinf_1j_pt100to200", "dy_m50toinf_1j_pt200to400",
         "dy_m50toinf_1j_pt400to600", "dy_m50toinf_1j_pt600toinf",

diff --git a/hbt/config/variables.py b/hbt/config/variables.py
@@ -188,6 +188,21 @@ def add_variables(config: od.Config) -> None:
         x_title=r"Jet 2 $p_{T}$",
     )
 
+    config.add_variable(
+        name="mu1_pt",
+        expression="Muon.pt[:,0]",
+        binning=(40, 0.0, 400.0),
+        unit="GeV",
+        x_title=r"Muon 1 $p_{T}$",
+    )
+    config.add_variable(
+        name="mu2_pt",
+        expression="Muon.pt[:,1]",
+        binning=(40, 0.0, 400.0),
+        unit="GeV",
+        x_title=r"Muon 2 $p_{T}$",
+    )
+
     # variables of interest
     config.add_variable(
         name="hh_mass",

diff --git a/hbt/weight/default.py b/hbt/weight/default.py
@@ -64,3 +64,9 @@ def default_init(self: WeightProducer) -> None:
     "normalization_only",
     cls_dict={"keep_weights": "normalization_weight"},
 )
+
+
+inclusive_only = default.derive(
+    "inclusive_only",
+    cls_dict={"keep_weights": "normalization_weight_incl"},
+)