add the bTagging_cff.py correction

cms-sw · Apr 8, 2024 · 9d40b1f · 9d40b1f
1 parent d3a211b
commit 9d40b1f
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/PhysicsTools/PatAlgos/python/recoLayer0/bTagging_cff.py b/PhysicsTools/PatAlgos/python/recoLayer0/bTagging_cff.py
@@ -62,6 +62,8 @@
   , 'pfParticleNetFromMiniAODAK4PuppiForwardTagInfos'
   , 'pfParticleNetFromMiniAODAK4CHSCentralTagInfos'
   , 'pfParticleNetFromMiniAODAK4CHSForwardTagInfos'
+  , 'pfNegativeParticleNetFromMiniAODAK4PuppiCentralTagInfos'
+  , 'pfNegativeParticleNetFromMiniAODAK4CHSCentralTagInfos'
   , 'pfParticleNetFromMiniAODAK8TagInfos'
 
   ]
@@ -303,6 +305,7 @@
 for disc in _pfNegativeParticleNetAK4JetTagsProbs:
     supportedBtagDiscr[disc] = [["pfNegativeParticleNetAK4TagInfos"]]
 # -----------------------------------
+# setup Run3 ParticleNet AK4
 from RecoBTag.ONNXRuntime.pfParticleNetFromMiniAODAK4_cff import _pfParticleNetFromMiniAODAK4PuppiCentralJetTagsProbs,_pfParticleNetFromMiniAODAK4PuppiCentralJetTagsMetaDiscr
 for disc in _pfParticleNetFromMiniAODAK4PuppiCentralJetTagsProbs:
     supportedBtagDiscr[disc] =  [["pfParticleNetFromMiniAODAK4PuppiCentralTagInfos"]]
@@ -323,6 +326,16 @@
     supportedBtagDiscr[disc] =  [["pfParticleNetFromMiniAODAK4CHSForwardTagInfos"]]
 for disc in _pfParticleNetFromMiniAODAK4CHSForwardJetTagsMetaDiscr:
     supportedMetaDiscr[disc] = _pfParticleNetFromMiniAODAK4CHSForwardJetTagsProbs
+# -----------------------------------
+# setup Negative Run3 ParticleNet AK4 (central only)
+from RecoBTag.ONNXRuntime.pfParticleNetFromMiniAODAK4_cff import _pfNegativeParticleNetFromMiniAODAK4PuppiCentralJetTagsProbs
+for disc in _pfNegativeParticleNetFromMiniAODAK4PuppiCentralJetTagsProbs:
+    supportedBtagDiscr[disc] =  [["pfNegativeParticleNetFromMiniAODAK4PuppiCentralTagInfos"]]
+from RecoBTag.ONNXRuntime.pfParticleNetFromMiniAODAK4_cff import _pfNegativeParticleNetFromMiniAODAK4CHSCentralJetTagsProbs
+for disc in _pfNegativeParticleNetFromMiniAODAK4CHSCentralJetTagsProbs:
+    supportedBtagDiscr[disc] =  [["pfNegativeParticleNetFromMiniAODAK4CHSCentralTagInfos"]]
+# -----------------------------------
+# setup Run3 ParticleNet AK8
 from RecoBTag.ONNXRuntime.pfParticleNetFromMiniAODAK8_cff import _pfParticleNetFromMiniAODAK8JetTagsProbs,_pfParticleNetFromMiniAODAK8JetTagsMetaDiscr
 for disc in _pfParticleNetFromMiniAODAK8JetTagsProbs:
     supportedBtagDiscr[disc] =  [["pfParticleNetFromMiniAODAK8TagInfos"]]
@@ -359,3 +372,12 @@
 for disc in _pfUnifiedParticleTransformerAK4JetTagsMetaDiscrs:
     supportedMetaDiscr[disc] = _pfUnifiedParticleTransformerAK4JetTagsProbs
 # -----------------------------------
+
+# -----------------------------------
+# setup Negative ParticleTransformer AK4
+from RecoBTag.ONNXRuntime.pfParticleTransformerAK4_cff import _pfNegativeParticleTransformerAK4JetTagsProbs
+# update supportedBtagDiscr
+for disc in _pfNegativeParticleTransformerAK4JetTagsProbs:
+    supportedBtagDiscr[disc] = [["pfNegativeParticleTransformerAK4TagInfos"]]
+
+