Merge branch 'run_dict' into run_dict_clib_run3

TopEFT · Nov 20, 2024 · 5b79f0a · 5b79f0a
2 parents da960fc + 978c741
commit 5b79f0a
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 4 deletions.
diff --git a/topcoffea/modules/run_dict.py b/topcoffea/modules/run_dict.py
@@ -0,0 +1,12 @@
+from topcoffea.modules.get_param_from_jsons import GetParam
+from topcoffea.modules.paths import topcoffea_path
+run_dict = GetParam(topcoffea_path("params/params.json"))
+
+def is_run2(year):
+    ''' Check if a year is in Run 2 '''
+    return year in run_dict("Run2")
+
+
+def is_run3(year):
+    ''' Check if a year is in Run 3 '''
+    return year in run_dict("Run3")
diff --git a/topcoffea/params/params.json b/topcoffea/params/params.json
@@ -9,6 +9,9 @@
     "lumi_2023": 17.794,
     "lumi_2023BPix": 9.451,
 
+    "Run2": ["2016", "2016APV", "2017", "2018"],
+    "Run3": ["2022", "2022EE", "2023", "2023BPix"],
+
 
     "mva_TTH_e_cut" : 0.90,
     "mva_TTH_m_cut" : 0.85,

diff --git a/topcoffea/params/run_dict.json b/topcoffea/params/run_dict.json