Merge pull request #70 from uhh-cms/run3_config

Run3 config

Author: apaasch

hbw/analysis/create_analysis.py

@@ -60,24 +60,57 @@ def create_hbw_analysis(
     #
 
     from hbw.config.config_run2 import add_config
+
     import cmsdb.campaigns.run2_2017_nano_v9
+    import cmsdb.campaigns.run3_2022_preEE_nano_v12
+    import cmsdb.campaigns.run3_2022_postEE_nano_v12
 
     campaign_run2_2017_nano_v9 = cmsdb.campaigns.run2_2017_nano_v9.campaign_run2_2017_nano_v9
+    campaign_run3_2022_preEE_nano_v12 = cmsdb.campaigns.run3_2022_preEE_nano_v12.campaign_run3_2022_preEE_nano_v12
+    campaign_run3_2022_preEE_nano_v12.x.EE = "pre"
+
+    campaign_run3_2022_postEE_nano_v12 = cmsdb.campaigns.run3_2022_postEE_nano_v12.campaign_run3_2022_postEE_nano_v12
+    campaign_run3_2022_postEE_nano_v12.x.EE = "post"
 
-    # default config
+    # default configs
     c17 = add_config(  # noqa
         analysis_inst,
         campaign_run2_2017_nano_v9.copy(),
         config_name="c17",
-        config_id=2,
+        config_id=17,
     )
-
-    # config with limited number of files
+    # configs with limited number of files
     l17 = add_config(  # noqa
         analysis_inst,
         campaign_run2_2017_nano_v9.copy(),
         config_name="l17",
-        config_id=12,
+        config_id=117,
+        limit_dataset_files=2,
+    )
+    c22pre = add_config(  # noqa
+        analysis_inst,
+        campaign_run3_2022_preEE_nano_v12.copy(),
+        config_name="c22pre",
+        config_id=2201,
+    )
+    l22pre = add_config(  # noqa
+        analysis_inst,
+        campaign_run3_2022_preEE_nano_v12.copy(),
+        config_name="l22pre",
+        config_id=12201,
+        limit_dataset_files=2,
+    )
+    c22post = add_config(  # noqa
+        analysis_inst,
+        campaign_run3_2022_postEE_nano_v12.copy(),
+        config_name="c22post",
+        config_id=2202,
+    )
+    l22post = add_config(  # noqa
+        analysis_inst,
+        campaign_run3_2022_postEE_nano_v12.copy(),
+        config_name="l22post",
+        config_id=12202,
         limit_dataset_files=2,
     )
 

hbw/calibration/default.py

@@ -59,7 +59,7 @@ def base_init(self: Calibrator) -> None:
     self.produces |= set(self.calibrators)
 
 
-default = base.derive("default", cls_dict=dict(skip_jecunc=True, bjet_regression=False))
+default = base.derive("default", cls_dict=dict(skip_jecunc=False, bjet_regression=False))
 skip_jecunc = base.derive("skip_jecunc", cls_dict=dict(skip_jecunc=True, bjet_regression=False))
 with_b_reg = base.derive("with_b_reg", cls_dict=dict(skip_jecunc=True, bjet_regression=True))
 full = base.derive("full", cls_dict=dict(skip_jecunc=False, bjet_regression=True))

hbw/calibration/jet.py

@@ -8,7 +8,7 @@
 
 import functools
 
-from columnflow.util import maybe_import, test_float
+from columnflow.util import maybe_import, try_float
 from columnflow.columnar_util import set_ak_column
 from columnflow.calibration.cms.jets import jec
 from columnflow.calibration import calibrator, Calibrator
@@ -60,7 +60,7 @@ def bjet_regression(
     default_jet_mask = (events.Jet.pt > 20)
     if self.btag_wp:
         btag_wp = self.btag_wp
-        if not test_float(self.btag_wp):
+        if not try_float(self.btag_wp):
             btag_wp = self.config_inst.x.btag_working_points.deepjet[self.btag_wp]
         default_jet_mask = default_jet_mask & (events.Jet.btagDeepFlavB > btag_wp)
 

hbw/config/config_run2.py

@@ -28,7 +28,7 @@ def add_cutflow_variables(config: od.Config) -> None:
     # Jets
     for i in range(4):
         # loose jets
-        for var in ("pt", "puId", "puIdDisc", "bRegRes", "bRegCorr", "qgl"):
+        for var in ("pt", "puId", "puIdDisc", "bRegRes", "bRegCorr"):
             quick_addvar(config, "LooseJet", i, var)
             quick_addvar(config, "Jet", i, var)
             quick_addvar(config, "JetPtBelow50", i, var)