Skip to content

Commit

Permalink
style: pre-commit fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
pre-commit-ci[bot] committed Sep 19, 2024
1 parent cbf8e00 commit a7a3ef6
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 6 deletions.
10 changes: 5 additions & 5 deletions src/HH4b/processors/GenSelection.py
Original file line number Diff line number Diff line change
Expand Up @@ -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[
Expand All @@ -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
Expand Down Expand Up @@ -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"""

Expand Down Expand Up @@ -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"""

Expand Down Expand Up @@ -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[
Expand Down
4 changes: 3 additions & 1 deletion src/HH4b/processors/bbbbSkimmer.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit a7a3ef6

Please sign in to comment.