FeatureTools/test/u0_cfg.py

0001 # Unit test configuration file for RecoBTagInfo producers:
0002 # Verify the use of unsubtracted jet map
0003 #
0004
0005 import FWCore.ParameterSet.Config as cms
0006
0007 process = cms.Process("TEST")
0008
0009 process.load('Configuration.Geometry.GeometryDB_cff')
0010 process.load('Configuration.StandardSequences.MagneticField_38T_cff')
0011 process.load('Configuration.StandardSequences.FrontierConditions_GlobalTag_cff')
0012 process.load("TrackingTools.TransientTrack.TransientTrackBuilder_cfi")
0013
0014 process.maxEvents = cms.untracked.PSet(
0015     input = cms.untracked.int32(2)
0016 )
0017
0018 process.source = cms.Source("PoolSource",
0019         fileNames = cms.untracked.vstring("/store/hidata/HIRun2023A/HIPhysicsRawPrime0/MINIAOD/PromptReco-v2/000/375/823/00000/8158260e-df3c-45a5-a121-55345a682a23.root")
0020 )
0021
0022 from Configuration.AlCa.GlobalTag import GlobalTag
0023 process.GlobalTag = GlobalTag(process.GlobalTag, '132X_dataRun3_Prompt_v4', '')
0024
0025 from RecoJets.JetProducers.ak4PFJets_cfi import ak4PFJets
0026 process.ak4PFJets = ak4PFJets.clone(rParam = 0.4, src = 'packedPFCandidates')
0027
0028 process.unsubJets = cms.EDProducer("JetMatcherDR",
0029     source = cms.InputTag("ak4PFJets"),
0030     matched = cms.InputTag("ak4PFJets")
0031 )
0032
0033 from RecoBTag.FeatureTools.pfDeepFlavourTagInfos_cfi import pfDeepFlavourTagInfos
0034 process.pfDeepFlavourTagInfos = pfDeepFlavourTagInfos.clone(
0035     jets = "ak4PFJets",
0036     unsubjet_map = "unsubJets",
0037     fallback_puppi_weight = True,
0038     fallback_vertex_association = True,
0039     puppi_value_map = "",
0040     secondary_vertices = "inclusiveCandidateSecondaryVertices",
0041     shallow_tag_infos = "pfDeepCSVTagInfos",
0042     vertex_associator = "",
0043     vertices = "offlineSlimmedPrimaryVertices"
0044 )
0045
0046 from RecoBTag.FeatureTools.pfDeepBoostedJetTagInfos_cfi import pfDeepBoostedJetTagInfos
0047 process.pfDeepBoostedJetTagInfos = pfDeepBoostedJetTagInfos.clone(
0048     jets = "ak4PFJets",
0049     unsubjet_map = "unsubJets",
0050     use_puppiP4 = False,
0051     puppi_value_map = "",
0052     secondary_vertices = "inclusiveCandidateSecondaryVertices",
0053     vertex_associator = "",
0054     vertices = "offlineSlimmedPrimaryVertices",
0055     pf_candidates = "packedPFCandidates"
0056 )
0057
0058 from RecoBTag.FeatureTools.pfParticleTransformerAK4TagInfos_cfi import pfParticleTransformerAK4TagInfos
0059 process.pfParticleTransformerAK4TagInfos = pfParticleTransformerAK4TagInfos.clone(
0060     jets = "ak4PFJets",
0061     unsubjet_map = "unsubJets",
0062     fallback_puppi_weight = True,
0063     fallback_vertex_association = True,
0064     puppi_value_map = "",
0065     secondary_vertices = "inclusiveCandidateSecondaryVertices",
0066     vertex_associator = "",
0067     vertices = "offlineSlimmedPrimaryVertices"
0068 )
0069
0070 process.p = cms.Path(process.ak4PFJets *
0071                      process.unsubJets *
0072                      process.pfDeepFlavourTagInfos *
0073                      process.pfDeepBoostedJetTagInfos *
0074                      process.pfParticleTransformerAK4TagInfos)