Fix code style issues with Black

rkansal47 · Aug 21, 2023 · c289357 · c289357
1 parent a9a4800
commit c289357
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 8 deletions.
diff --git a/src/HHbbVV/processors/bbVVSkimmer.py b/src/HHbbVV/processors/bbVVSkimmer.py
@@ -118,7 +118,13 @@ class bbVVSkimmer(processor.ProcessorABC):
         min_branches.append(f"DijetMass_{shift}")
 
     def __init__(
-        self, xsecs={}, save_ak15=False, save_systematics=True, inference=True, save_all=True,vbf_search = False
+        self,
+        xsecs={},
+        save_ak15=False,
+        save_systematics=True,
+        inference=True,
+        save_all=True,
+        vbf_search=False,
     ):
         super(bbVVSkimmer, self).__init__()
 
@@ -133,7 +139,7 @@ def __init__(
 
         # save all branches or only necessary ones
         self._save_all = save_all
-        
+
         # search for VBF production events
         self._vbf_search = vbf_search
 
@@ -779,11 +785,7 @@ def getVBFVars(
         )  # this might not work due to types
 
         # Apply base masks, sort, and calculate vbf dijet (jj) cuts
-        vbfJets_mask = (
-            ak4_jet_mask
-            & electron_muon_overlap_mask
-            & fatjet_overlap_mask
-        )
+        vbfJets_mask = ak4_jet_mask & electron_muon_overlap_mask & fatjet_overlap_mask
         vbfJets = jets[vbfJets_mask]
 
         vbfJets_sorted_pt = vbfJets[ak.argsort(vbfJets.pt, ascending=False)]

diff --git a/src/run_utils.py b/src/run_utils.py
@@ -182,4 +182,6 @@ def parse_common_args(parser):
     add_bool_arg(parser, "save-systematics", default=False, help="save systematic variations")
     add_bool_arg(parser, "save-all", default=True, help="save all branches")
     add_bool_arg(parser, "inference", default=True, help="run inference for ak8 jets")
-    add_bool_arg(parser, "vbf-search", default=False, help="run selections for VBF production search")
+    add_bool_arg(
+        parser, "vbf-search", default=False, help="run selections for VBF production search"
+    )