diff --git a/PhysicsTools/NanoAOD/python/custom_jme_cff.py b/PhysicsTools/NanoAOD/python/custom_jme_cff.py index d82138ab7752a..8c57d9ff5683b 100644 --- a/PhysicsTools/NanoAOD/python/custom_jme_cff.py +++ b/PhysicsTools/NanoAOD/python/custom_jme_cff.py @@ -662,9 +662,26 @@ def ReclusterAK4PuppiJets(proc, recoJA, runOnMC): proc.jetPuppiTable.variables.phoMultiplicity = PFJETVARS.phoMultiplicity # - # Add variables for pileup jet ID studies. + # Add Pileup Jet ID for Puppi jets # + from RecoJets.JetProducers.PileupJetID_cfi import pileupJetIdPuppi + pileupJetIdName = "pileupJetId{}".format(jetName) + setattr(proc, pileupJetIdName, pileupJetIdPuppi.clone( + jets = "updatedJetsPuppi", + srcConstituentWeights = "packedpuppi", + vertexes = "offlineSlimmedPrimaryVertices", + inputIsCorrected=True, + applyJec=False + ) + ) + proc.jetPuppiTask.add(getattr(proc, pileupJetIdName)) + proc.updatedJetsPuppiWithUserData.userFloats.puIdDisc = cms.InputTag(pileupJetIdName+':fullDiscriminant') + proc.jetPuppiTable.variables.puIdDisc = Var("userFloat('puIdDisc')", float, doc="Pileup ID BDT discriminant with 133X Winter24 PuppiV18 training",precision=10) + + # + # Add variables for pileup jet ID studies. + # proc = AddPileUpJetIDVars(proc, jetName = jetName, jetSrc = "updatedJetsPuppi",