fix str

LPC-HH · Sep 19, 2024 · c98d4c6 · c98d4c6
2 parents be27091 + a7a3ef6
commit c98d4c6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git 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
@@ -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