diff --git a/src/HH4b/processors/GenSelection.py b/src/HH4b/processors/GenSelection.py index 41118290..6075c62f 100644 --- a/src/HH4b/processors/GenSelection.py +++ b/src/HH4b/processors/GenSelection.py @@ -44,7 +44,7 @@ def gen_selection_HHbbbb_simplified( fatjets: FatJetArray, # noqa: ARG001 selection_args: list, # noqa: ARG001 skim_vars: dict, - fatjet_str: str, # noqa: ARG001 + fatjet_str: str, # noqa: ARG001 ): """Simplified gen selection""" higgs = events.GenPart[ @@ -209,7 +209,7 @@ def gen_selection_Top( fatjets: FatJetArray, selection_args: list, # noqa: ARG001 skim_vars: dict, - fatjet_str: str, # noqa: ARG001 + fatjet_str: str, ): """Get Hadronic Top and children information""" @@ -285,7 +285,7 @@ def gen_selection_V( fatjets: FatJetArray, selection_args: list, # noqa: ARG001 skim_vars: dict, - fatjet_str: str, # noqa: ARG001 + fatjet_str: str, ): """Get W/Z and children information""" vs = events.GenPart[ diff --git a/src/HH4b/processors/bbbbSkimmer.py b/src/HH4b/processors/bbbbSkimmer.py index b15e8fc1..9754caf4 100644 --- a/src/HH4b/processors/bbbbSkimmer.py +++ b/src/HH4b/processors/bbbbSkimmer.py @@ -457,7 +457,9 @@ def process(self, events: ak.Array): # match fatjets_xbb # vars_dict = gen_selection_dict[d](events, jets, fatjets_xbb, selection_args, P4, "bbFatJet") # match fatjets - vars_dict = gen_selection_dict[d](events, jets, fatjets, selection_args, P4, "ak8FatJet") + vars_dict = gen_selection_dict[d]( + events, jets, fatjets, selection_args, P4, "ak8FatJet" + ) genVars = {**genVars, **vars_dict} # remove unnecessary ak4 gen variables for signal region