From a7a3ef6a75af9763d47f7fb6ef898b66e56fd1a6 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 19 Sep 2024 21:14:01 +0000 Subject: [PATCH] style: pre-commit fixes --- src/HH4b/processors/GenSelection.py | 10 +++++----- src/HH4b/processors/bbbbSkimmer.py | 4 +++- 2 files changed, 8 insertions(+), 6 deletions(-) diff --git a/src/HH4b/processors/GenSelection.py b/src/HH4b/processors/GenSelection.py index cebc7b23..3aaf032a 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[ @@ -65,7 +65,7 @@ def gen_selection_HHbbbb( fatjets: FatJetArray, selection_args: list, skim_vars: dict, - fatjet_str: "bbFatJet", + fatjet_str: bbFatJet, ): """Gets HH, bb 4-vectors""" # finding the two gen higgs @@ -157,7 +157,7 @@ def gen_selection_Hbb( fatjets: FatJetArray, selection_args: list, # noqa: ARG001 skim_vars: dict, - fatjet_str: "bbFatJet", + fatjet_str: bbFatJet, ): """Gets H, bb, 4-vectors + Higgs children information""" @@ -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