merge

rkansal47 · Nov 11, 2023 · e6b2642 · e6b2642
2 parents 82001a9 + 3d95d58
commit e6b2642
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/src/HHbbVV/postprocessing/postprocessing.py b/src/HHbbVV/postprocessing/postprocessing.py
@@ -190,7 +190,6 @@ def get_nonres_vbf_selection_regions(
                 "ak8FatJetEta1": [-2.4, 2.4],
                 "DijetdEta": [0, 2.0],
                 "DijetdPhi": [2.6, 10000],
-                # add more
             },
             label="Pass",
         ),
@@ -206,7 +205,6 @@ def get_nonres_vbf_selection_regions(
                 "ak8FatJetEta1": [-2.4, 2.4],
                 "DijetdEta": [0, 2.0],
                 "DijetdPhi": [2.6, 10000],
-                # add more
             },
             label="Fail",
         ),
@@ -352,7 +350,7 @@ def get_res_selection_regions(
 weight_shifts = {
     "pileup": Syst(samples=nonres_sig_keys + res_sig_keys + bg_keys, label="Pileup"),
     # "PDFalphaS": Syst(samples=nonres_sig_keys, label="PDF"),
-    # "QCDscale": Syst(samples=nonres_sig_keys, label="QCDscale"), # commented out
+    # "QCDscale": Syst(samples=nonres_sig_keys, label="QCDscale"),
     "ISRPartonShower": Syst(samples=nonres_sig_keys_ggf + ["V+Jets"], label="ISR Parton Shower"),
     "FSRPartonShower": Syst(samples=nonres_sig_keys_ggf + ["V+Jets"], label="FSR Parton Shower"),
     "L1EcalPrefiring": Syst(
@@ -410,14 +408,9 @@ def main(args):
         pt_labels = ["JES", "JER"]
         m_labels = ["JMS", "JMR"]
         for key, df in events_dict.items():
-            # for each JES,JER, and JMS JMR correction
-            ptlabel = "_JES_up"
-            mlabel = "_JMS_down"
-
             bb_mask = (
                 df[("ak8FatJetParticleNetMD_Txbb", 0)] > df[("ak8FatJetParticleNetMD_Txbb", 1)]
             )
-
             _add_vbf_columns(df, bb_mask, ptlabel="", mlabel="")
 
             if key == "Data":

diff --git a/src/HHbbVV/processors/bbVVSkimmer.py b/src/HHbbVV/processors/bbVVSkimmer.py
@@ -631,6 +631,9 @@ def process(self, events: ak.Array):
             if self._systematics:
                 systematics += list(weights.variations)
 
+            single_weight_pileup = weights.partial_weight(["single_weight_pileup"])
+            add_selection("single_weight_pileup", (single_weight_pileup <= 4), *selection_args)
+
             # TODO: need to be careful about the sum of gen weights used for the LHE/QCDScale uncertainties
             logger.debug("weights ", weights._weights.keys())