Allow adding to existing tauEmbedder

cms-sw · Jun 28, 2019 · fd4728f · fd4728f
1 parent e6888e9
commit fd4728f
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/RecoTauTag/RecoTau/python/tools/runTauIdMVA.py b/RecoTauTag/RecoTau/python/tools/runTauIdMVA.py
@@ -1043,11 +1043,17 @@ def runTauID(self):
 
         ##
         if self.debug: print('Embedding new TauIDs into \"'+self.updatedTauName+'\"')
-        embedID = self.cms.EDProducer("PATTauIDEmbedder",
-            src = self.cms.InputTag('slimmedTaus'),
-            tauIDSources = tauIDSources
-        )
-        setattr(self.process, self.updatedTauName, embedID)
+        if not hasattr(self.process, self.updatedTauName):
+            embedID = self.cms.EDProducer("PATTauIDEmbedder",
+               src = self.cms.InputTag('slimmedTaus'),
+               tauIDSources = tauIDSources
+            )
+            setattr(self.process, self.updatedTauName, embedID)
+        else: #assume same type
+            tauIDSources = self.cms.PSet(
+                getattr(self.process, self.updatedTauName).tauIDSources,
+                tauIDSources)
+            getattr(self.process, self.updatedTauName).tauIDSources = tauIDSources
 
 
     def processDeepProducer(self, producer_name, tauIDSources, workingPoints_):