Fix btvNano.

Author: hqucms

PhysicsTools/NanoAOD/python/custom_btv_cff.py

@@ -65,8 +65,10 @@ def update_jets_AK4(process):
     process.updatedPatJetsTransientCorrectedPuppiWithDeepInfo.tagInfoSources.append(cms.InputTag("pfUnifiedParticleTransformerAK4TagInfosPuppiWithDeepInfo"))
     process.updatedPatJetsTransientCorrectedPuppiWithDeepInfo.addTagInfos = cms.bool(True)
 
-    
-    
+    # Fix ParticleNetFromMiniAOD input when slimmedTaus is updated
+    from PhysicsTools.NanoAOD.nano_cff import _fixPNetInputCollection
+    _fixPNetInputCollection(process)
+
     return process
 
 def update_jets_AK8(process):

PhysicsTools/NanoAOD/python/custom_jme_cff.py

@@ -1509,6 +1509,12 @@ def addAK4JetTasks(proc, addAK4CHSJetTasks, addAK4PuppiJetTasks):
     addAK4PuppiJetTasks = jmeNano_addAK4JetTasks_switch.addAK4Puppi_switch,
   )
 
+  ###########################################################################
+  # Fix ParticleNetFromMiniAOD input when slimmedTaus is updated
+  ###########################################################################
+  from PhysicsTools.NanoAOD.nano_cff import _fixPNetInputCollection
+  _fixPNetInputCollection(process)
+
   ###########################################################################
   # Save Maximum of Pt Hat Max
   ###########################################################################

PhysicsTools/NanoAOD/python/nano_cff.py

@@ -93,6 +93,14 @@
 nanoSequenceMC.insert(nanoSequenceFS.index(nanoSequenceCommon)+1,nanoSequenceOnlyFullSim)
 
 
+def _fixPNetInputCollection(process):
+    # fix circular module dependency in ParticleNetFromMiniAOD TagInfos when slimmedTaus is updated
+    if hasattr(process, 'slimmedTaus'):
+        for mod in process.producers.keys():
+            if 'ParticleNetFromMiniAOD' in mod and 'TagInfos' in mod:
+                getattr(process, mod).taus = 'slimmedTaus::@skipCurrentProcess'
+
+
 # modifier which adds new tauIDs
 import RecoTauTag.RecoTau.tools.runTauIdMVA as tauIdConfig
 def nanoAOD_addTauIds(process, idsToRun=[], addPNetCHS=False, addUParTPuppi=False):
@@ -173,11 +181,7 @@ def nanoAOD_addTauIds(process, idsToRun=[], addPNetCHS=False, addUParTPuppi=Fals
         process.slimmedTaus = getattr(process, updatedTauName).clone()
         process.tauTask.replace(getattr(process, updatedTauName), process.slimmedTaus)
         delattr(process, updatedTauName)
-
-        # fix circular module dependency in :
-        for mod in process.producers.keys():
-            if 'ParticleNetFromMiniAOD' in mod and 'TagInfos' in mod:
-                getattr(process, mod).taus = 'slimmedTaus::@skipCurrentProcess'
+        _fixPNetInputCollection(process)
 
     return process