Remove old b-taggers for AK8 subjets. Add DeepJet and UParT.

Author: nurfikri89

PhysicsTools/PatAlgos/python/slimming/applySubstructure_cff.py

@@ -63,15 +63,27 @@ def applySubstructure( process, postfix="" ) :
         jetCorrections = ('AK8PFPuppi', ['L2Relative', 'L3Absolute'], 'None'),
         getJetMCFlavour = False # jet flavor disabled
     )
+
     ## PATify soft drop subjets
+    from RecoBTag.ONNXRuntime.pfUnifiedParticleTransformerAK4_cff import _pfUnifiedParticleTransformerAK4JetTagsAll as pfUnifiedParticleTransformerAK4JetTagsAll
+    _btagDiscriminatorsSubjets = cms.PSet(
+      names=cms.vstring(
+        'pfDeepFlavourJetTags:probb',
+        'pfDeepFlavourJetTags:probbb',
+        'pfDeepFlavourJetTags:problepb',
+        'pfUnifiedParticleTransformerAK4DiscriminatorsJetTags:BvsAll'
+      )
+      # + pfUnifiedParticleTransformerAK4JetTagsAll
+    )
+
     addJetCollection(
         process,
         postfix=postfix,
         labelName = 'AK8PFPuppiSoftDropSubjets',
         jetSource = cms.InputTag('ak8PFJetsPuppiSoftDrop'+postfix,'SubJets'),
         algo = 'ak',  # needed for subjet flavor clustering
         rParam = 0.8, # needed for subjet flavor clustering
-        btagDiscriminators = ['pfDeepCSVJetTags:probb', 'pfDeepCSVJetTags:probbb', 'pfCombinedInclusiveSecondaryVertexV2BJetTags','pfCombinedMVAV2BJetTags'],
+        btagDiscriminators = _btagDiscriminatorsSubjets.names.value(),
         jetCorrections = ('AK4PFPuppi', ['L2Relative', 'L3Absolute'], 'None'),
         explicitJTA = True,  # needed for subjet b tagging
         svClustering = True, # needed for subjet b tagging
@@ -80,14 +92,6 @@ def applySubstructure( process, postfix="" ) :
         groomedFatJets=cms.InputTag('ak8PFJetsPuppiSoftDrop') # needed for subjet flavor clustering
     )
 
-    from Configuration.Eras.Modifier_run3_common_cff import run3_common
-    run3_common.toModify(process.patJetsAK8PFPuppiSoftDropSubjets,
-                         discriminatorSources = cms.VInputTag(
-                            cms.InputTag("pfDeepCSVJetTagsAK8PFPuppiSoftDropSubjets","probb"),
-                            cms.InputTag("pfDeepCSVJetTagsAK8PFPuppiSoftDropSubjets","probbb")
-                         )
-    )
-
     # add groomed ECFs and N-subjettiness to soft dropped pat::Jets for fat jets and subjets
     process.load('RecoJets.JetProducers.ECF_cff')
     addToProcessAndTask('nb1AK8PuppiSoftDrop'+postfix, process.ecfNbeta1.clone(src = cms.InputTag("ak8PFJetsPuppiSoftDrop"+postfix), cuts = cms.vstring('', '', 'pt > 250')), process, task)