upup

Author: farakiko

boostedhiggs/hwwprocessor.py

@@ -89,37 +89,29 @@ def __init__(
     ):
         self._year = year
         self._yearmod = yearmod
-        self._channels = channels
+        self._channels = channels or ["ele", "mu"]
+
         self._systematics = systematics
         self._getLPweights = getLPweights
         self._uselooselep = uselooselep
         self._fakevalidation = fakevalidation
-
         self._apply_trigger = not no_trigger
         self._apply_selection = not no_selection
-
         self._output_location = output_location
+        self._inference = inference
 
-        # trigger paths
-        with importlib.resources.path("boostedhiggs.data", "triggers.json") as path:
-            with open(path, "r") as f:
-                self._HLTs = json.load(f)[self._year]
+        def load_json_config(file_name):
+            with importlib.resources.path("boostedhiggs.data", file_name) as path:
+                with open(path, "r") as f:
+                    return json.load(f)[self._year]
 
-        # https://twiki.cern.ch/twiki/bin/view/CMS/MissingETOptionalFiltersRun2
-        with importlib.resources.path("boostedhiggs.data", "metfilters.json") as path:
-            with open(path, "r") as f:
-                self._metfilters = json.load(f)[self._year]
+        self._HLTs = load_json_config("triggers.json")
+        self._metfilters = load_json_config("metfilters.json")
 
-        if self._year == "2018":
-            self.dataset_per_ch = {
-                "ele": "EGamma",
-                "mu": "SingleMuon",
-            }
-        else:
-            self.dataset_per_ch = {
-                "ele": "SingleElectron",
-                "mu": "SingleMuon",
-            }
+        self.dataset_per_ch = {
+            "ele": "EGamma" if self._year == "2018" else "SingleElectron",
+            "mu": "SingleMuon",
+        }
 
         self.jecs = {
             "JES": "JES_jes",
@@ -139,9 +131,7 @@ def __init__(
             "JES_Total": "JES_Total",
         }
 
-        # for tagger inference
-        self._inference = inference
-        self.tagger_resources_path = str(pathlib.Path(__file__).parent.resolve()) + "/tagger_resources/"
+        self.tagger_resources_path = os.path.join(str(pathlib.Path(__file__).parent.resolve()), "tagger_resources")
 
     @property
     def accumulator(self):