bug fix

rkansal47 · Sep 20, 2023 · a3c5a60 · a3c5a60
1 parent fb701cf
commit a3c5a60
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/HHbbVV/postprocessing/postprocessing.py b/src/HHbbVV/postprocessing/postprocessing.py
@@ -705,7 +705,8 @@ def apply_weights(
     for sample in events_dict:
         events = events_dict[sample]
         if sample == data_key:
-            events[weight_key] = events["weight"]
+            if weight_key not in events:
+                events[weight_key] = events["weight"]
         elif f"{weight_key}_noTrigEffs" not in events:
             fj_trigeffs = ak8TrigEffsLookup(
                 events["ak8FatJetParticleNetMD_Txbb"].values,