METPUSubtraction/test/testDeepMETSonic_cfg.py

0001 import FWCore.ParameterSet.Config as cms
0002 from RecoMET.METPUSubtraction.deepMetSonicProducer_cff import sonic_deepmet
0003
0004 from Configuration.ProcessModifiers.enableSonicTriton_cff import enableSonicTriton
0005 process = cms.Process('DeepMET',enableSonicTriton)
0006
0007 # import of standard configurations
0008 process.load('Configuration.StandardSequences.Services_cff')
0009 process.load('Configuration.EventContent.EventContent_cff')
0010 process.load('Configuration.StandardSequences.EndOfProcess_cff')
0011 process.load('Configuration.StandardSequences.FrontierConditions_GlobalTag_cff')
0012
0013 process.maxEvents = cms.untracked.PSet(
0014     input=cms.untracked.int32(10)
0015 )
0016
0017 # Input source
0018 process.source = cms.Source("PoolSource",
0019                             fileNames=cms.untracked.vstring(
0020                                 '/store/mc/RunIISummer20UL18MiniAOD/DYJetsToLL_M-50_TuneCP5_13TeV-madgraphMLM-pythia8/MINIAODSIM/106X_upgrade2018_realistic_v11_L1v1-v1/260000/03794341-C401-CC45-B5FC-D11264E449CE.root'
0021                             ),
0022                             )
0023
0024 process.load("HeterogeneousCore.SonicTriton.TritonService_cff")
0025 process.TritonService.verbose = False
0026 #process.TritonService.fallback.useDocker = True
0027 process.TritonService.fallback.verbose = False
0028 # uncomment this part if there is one server running at 0.0.0.0 with grpc port 8001
0029 #process.TritonService.servers.append(
0030 #    cms.PSet(
0031 #        name = cms.untracked.string("default"),
0032 #        address = cms.untracked.string("0.0.0.0"),
0033 #        port = cms.untracked.uint32(8021),
0034 #    )
0035 #)
0036
0037
0038 process.deepMETProducer = sonic_deepmet.clone(
0039 )
0040 process.deepMETProducer.Client.verbose = cms.untracked.bool(True)
0041
0042 process.p = cms.Path()
0043 process.p += process.deepMETProducer
0044
0045 process.output = cms.OutputModule("PoolOutputModule",
0046                                   outputCommands=cms.untracked.vstring(
0047                                       'keep *'),
0048                                   fileName=cms.untracked.string(
0049                                       "DeepMETTestSonic.root"
0050                                       )
0051                                   )
0052 process.outpath  = cms.EndPath(process.output)
0053
0054 process.options = cms.untracked.PSet(
0055     wantSummary = cms.untracked.bool( True ),
0056     numberOfThreads = cms.untracked.uint32( 1 ),
0057     numberOfStreams = cms.untracked.uint32( 0 ),
0058     sizeOfStackForThreadsInKB = cms.untracked.uint32( 10*1024 )
0059 )