ONNXRuntime/test/test_particlenet_from_miniaod_ak4chs_cfg.py

0001 ### CMSSW command line parameter parser
0002 import FWCore.ParameterSet.Config as cms
0003
0004
0005 process = cms.Process("test")
0006
0007 # Message Logger settings
0008 process.load("FWCore.MessageService.MessageLogger_cfi")
0009 process.MessageLogger.cerr.FwkReport.reportEvery = 250
0010 process.load("RecoBTag.ONNXRuntime.pfParticleNetFromMiniAODAK4_cff")
0011
0012 process.maxEvents = cms.untracked.PSet(
0013     input = cms.untracked.int32(100)
0014 )
0015
0016 process.options = cms.untracked.PSet(
0017     allowUnscheduled = cms.untracked.bool(True),
0018     wantSummary      = cms.untracked.bool(True),
0019     numberOfThreads  = cms.untracked.uint32(8),
0020     numberOfStreams = cms.untracked.uint32(0)
0021 )
0022
0023 process.source = cms.Source("PoolSource",
0024     fileNames = cms.untracked.vstring("/store/mc/RunIISummer20UL18MiniAODv2/TTToSemiLeptonic_TuneCP5_13TeV-powheg-pythia8/MINIAODSIM/106X_upgrade2018_realistic_v16_L1v1-v2/130000/25BF763A-BF41-E242-86A2-5E0BE8EF605C.root")
0025 )
0026
0027 process.load('Configuration.StandardSequences.FrontierConditions_GlobalTag_cff')
0028 process.load('Configuration.StandardSequences.GeometryDB_cff')
0029 process.load('Configuration.StandardSequences.MagneticField_cff')
0030 process.load('TrackingTools.TransientTrack.TransientTrackBuilder_cfi')
0031
0032 from Configuration.AlCa.GlobalTag import GlobalTag
0033 process.GlobalTag = GlobalTag(process.GlobalTag, 'auto:phase1_2018_realistic', '')
0034
0035
0036 from PhysicsTools.PatAlgos.producersLayer1.jetUpdater_cfi import updatedPatJets
0037 process.slimmedJetsUpdated = updatedPatJets.clone(
0038     jetSource = "slimmedJets",
0039     addJetCorrFactors = False,
0040     discriminatorSources = cms.VInputTag(
0041         cms.InputTag("pfParticleNetFromMiniAODAK4CHSCentralJetTags:probb"),
0042         cms.InputTag("pfParticleNetFromMiniAODAK4CHSCentralJetTags:probc"),
0043         cms.InputTag("pfParticleNetFromMiniAODAK4CHSCentralJetTags:probuds"),
0044         cms.InputTag("pfParticleNetFromMiniAODAK4CHSCentralJetTags:probg"),
0045         cms.InputTag("pfParticleNetFromMiniAODAK4CHSCentralJetTags:probmu"),
0046         cms.InputTag("pfParticleNetFromMiniAODAK4CHSCentralJetTags:probele"),
0047         cms.InputTag("pfParticleNetFromMiniAODAK4CHSCentralJetTags:probtaup1h0p"),
0048         cms.InputTag("pfParticleNetFromMiniAODAK4CHSCentralJetTags:probtaup1h1p"),
0049         cms.InputTag("pfParticleNetFromMiniAODAK4CHSCentralJetTags:probtaup1h2p"),
0050         cms.InputTag("pfParticleNetFromMiniAODAK4CHSCentralJetTags:probtaup3h0p"),
0051         cms.InputTag("pfParticleNetFromMiniAODAK4CHSCentralJetTags:probtaup3h1p"),
0052         cms.InputTag("pfParticleNetFromMiniAODAK4CHSCentralJetTags:probtaum1h0p"),
0053         cms.InputTag("pfParticleNetFromMiniAODAK4CHSCentralJetTags:probtaum1h1p"),
0054         cms.InputTag("pfParticleNetFromMiniAODAK4CHSCentralJetTags:probtaum1h2p"),
0055         cms.InputTag("pfParticleNetFromMiniAODAK4CHSCentralJetTags:probtaum3h0p"),
0056         cms.InputTag("pfParticleNetFromMiniAODAK4CHSCentralJetTags:probtaum3h1p"),
0057         cms.InputTag("pfParticleNetFromMiniAODAK4CHSCentralJetTags:ptcorr"),
0058         cms.InputTag("pfParticleNetFromMiniAODAK4CHSCentralJetTags:ptnu"),
0059         cms.InputTag("pfParticleNetFromMiniAODAK4CHSCentralJetTags:ptreshigh"),
0060         cms.InputTag("pfParticleNetFromMiniAODAK4CHSCentralJetTags:ptreslow"),
0061
0062         cms.InputTag("pfParticleNetFromMiniAODAK4CHSForwardJetTags:probq"),
0063         cms.InputTag("pfParticleNetFromMiniAODAK4CHSForwardJetTags:probg"),
0064         cms.InputTag("pfParticleNetFromMiniAODAK4CHSForwardJetTags:ptcorr"),
0065         cms.InputTag("pfParticleNetFromMiniAODAK4CHSForwardJetTags:ptnu"),
0066         cms.InputTag("pfParticleNetFromMiniAODAK4CHSForwardJetTags:ptreshigh"),
0067         cms.InputTag("pfParticleNetFromMiniAODAK4CHSForwardJetTags:ptreslow"),
0068
0069     )
0070 )
0071
0072
0073 process.path = cms.Path(
0074     process.slimmedJetsUpdated,process.pfParticleNetFromMiniAODAK4CHSTask
0075 )
0076
0077 process.output = cms.OutputModule( "PoolOutputModule",
0078     fileName = cms.untracked.string( "test.root" ),
0079     compressionAlgorithm = cms.untracked.string('LZMA'),
0080     compressionLevel = cms.untracked.int32(4),
0081     eventAutoFlushCompressedSize = cms.untracked.int32(31457280),
0082     dataset = cms.untracked.PSet(
0083         dataTier = cms.untracked.string( 'RECO' ),
0084         filterName = cms.untracked.string( '' )
0085     ),
0086     overrideBranchesSplitLevel = cms.untracked.VPSet(),
0087     outputCommands = cms.untracked.vstring(
0088         'drop *',
0089         'keep *_*slimmedJetsUpdated*_*_*',
0090     )
0091 )
0092
0093 process.endpath = cms.EndPath(process.output);
0094 process.schedule = cms.Schedule(process.path,process.endpath);
0095