ONNXRuntime/test/test_unifiedpartak4_cfg.py

0001
0002 import FWCore.ParameterSet.Config as cms
0003 from PhysicsTools.PatAlgos.tools.helpers import getPatAlgosToolsTask
0004
0005 process = cms.Process("PATtest")
0006
0007 ## MessageLogger
0008 process.load("FWCore.MessageLogger.MessageLogger_cfi")
0009
0010 ## Options and Output Report
0011 process.options = cms.untracked.PSet( wantSummary = cms.untracked.bool(True) )
0012
0013 ## Source
0014 process.source = cms.Source("PoolSource",
0015     fileNames = cms.untracked.vstring()
0016 )
0017 ## Maximal Number of Events
0018 process.maxEvents = cms.untracked.PSet( input = cms.untracked.int32(1000))
0019
0020 ## Geometry and Detector Conditions (needed for a few patTuple production steps)
0021 process.load("Configuration.Geometry.GeometryRecoDB_cff")
0022 process.load("Configuration.StandardSequences.FrontierConditions_GlobalTag_cff")
0023 from Configuration.AlCa.GlobalTag import GlobalTag
0024 process.GlobalTag = GlobalTag(process.GlobalTag, 'auto:run2_mc')
0025 process.load("Configuration.StandardSequences.MagneticField_cff")
0026
0027 ## Output Module Configuration (expects a path 'p')
0028 from PhysicsTools.PatAlgos.patEventContent_cff import patEventContentNoCleaning
0029 process.out = cms.OutputModule("PoolOutputModule",
0030                                fileName = cms.untracked.string('patTuple.root'),
0031                                outputCommands = cms.untracked.vstring('drop *', *patEventContentNoCleaning )
0032                                )
0033
0034 patAlgosToolsTask = getPatAlgosToolsTask(process)
0035 process.outpath = cms.EndPath(process.out, patAlgosToolsTask)
0036
0037 ## and add them to the event content
0038 from PhysicsTools.PatAlgos.tools.jetTools import updateJetCollection
0039
0040 updateJetCollection(
0041    process,
0042    jetSource = cms.InputTag('slimmedJetsPuppi'),
0043    pvSource = cms.InputTag('offlineSlimmedPrimaryVertices'),
0044    svSource = cms.InputTag('slimmedSecondaryVertices'),
0045    jetCorrections = ('AK4PFchs', cms.vstring(['L1FastJet', 'L2Relative', 'L3Absolute']), 'None'),
0046    btagDiscriminators = [
0047       'pfDeepFlavourJetTags:probb',
0048       'pfDeepFlavourJetTags:probbb',
0049       'pfDeepFlavourJetTags:problepb',
0050       'pfDeepFlavourJetTags:probc',
0051       'pfDeepFlavourJetTags:probuds',
0052       'pfDeepFlavourJetTags:probg',
0053       'pfUnifiedParticleTransformerAK4JetTags:probb',
0054       'pfUnifiedParticleTransformerAK4JetTags:probbb',
0055       'pfUnifiedParticleTransformerAK4JetTags:problepb',
0056       'pfUnifiedParticleTransformerAK4JetTags:probc',
0057       'pfUnifiedParticleTransformerAK4JetTags:probs',
0058       'pfUnifiedParticleTransformerAK4JetTags:probu',
0059       'pfUnifiedParticleTransformerAK4JetTags:probd',
0060       'pfUnifiedParticleTransformerAK4JetTags:probg',
0061       'pfParticleTransformerAK4JetTags:probb',
0062       'pfParticleTransformerAK4JetTags:probbb',
0063       'pfParticleTransformerAK4JetTags:problepb',
0064       'pfParticleTransformerAK4JetTags:probc',
0065       'pfParticleTransformerAK4JetTags:probuds',
0066       'pfParticleTransformerAK4JetTags:probg',
0067       'pfParticleNetFromMiniAODAK4PuppiCentralJetTags:probb',
0068       'pfParticleNetFromMiniAODAK4PuppiCentralJetTags:probc',
0069       'pfParticleNetFromMiniAODAK4PuppiCentralJetTags:probuds',
0070       'pfParticleNetFromMiniAODAK4PuppiCentralJetTags:probg',
0071       ]
0072    )
0073
0074 from PhysicsTools.PatAlgos.patInputFiles_cff import filesRelValTTbarPileUpMINIAODSIM
0075
0076 process.source.fileNames = filesRelValTTbarPileUpMINIAODSIM
0077 process.source.fileNames = cms.untracked.vstring('/store/mc/Run3Summer23BPixMiniAODv4/TTtoLNu2Q_TuneCP5_13p6TeV_powheg-pythia8/MINIAODSIM/130X_mcRun3_2023_realistic_postBPix_v2-v3/2520000/00488681-4f49-4bdc-89e6-198da9e42a17.root')
0078
0079 process.maxEvents.input = 1000
0080
0081 from Configuration.EventContent.EventContent_cff import MINIAODSIMEventContent
0082 process.out.outputCommands = MINIAODSIMEventContent.outputCommands
0083 process.out.outputCommands.append('keep *_selectedUpdatedPatJets*_*_*')
0084 process.out.outputCommands.append('keep *_pfDeepFlavourTagInfos*_*_*')
0085 process.out.outputCommands.append('keep *_pfDeepFlavourJetTags*_*_*')
0086 process.out.outputCommands.append('keep *_updatedPatJets*_*_*')
0087
0088 process.out.fileName = 'test_deep_unifiedpartak4_MINIAODSIM.root'