From 8764b76aea4113bf8258a45aee6fcbe01ec6bf8f Mon Sep 17 00:00:00 2001
From: Yuan CHAO <yuanchao@cern.ch>
Date: Thu, 17 Nov 2022 16:30:29 +0100
Subject: [PATCH] Mask old b-taggers and keep only DNN taggers with Era
 modifier for run3.

---
 RecoBTag/Configuration/python/RecoBTag_cff.py | 29 +++++++++++++++++--
 1 file changed, 26 insertions(+), 3 deletions(-)

diff --git a/RecoBTag/Configuration/python/RecoBTag_cff.py b/RecoBTag/Configuration/python/RecoBTag_cff.py
index e1f096c7a5774..30fad5b537327 100644
--- a/RecoBTag/Configuration/python/RecoBTag_cff.py
+++ b/RecoBTag/Configuration/python/RecoBTag_cff.py
@@ -92,7 +92,30 @@
 
 ## modifying b-tagging task in Run3 adding ParticleNet inferece
 from Configuration.Eras.Modifier_run3_common_cff import run3_common
-_pfBTaggingTask_particleNet = pfBTaggingTask.copy()
-_pfBTaggingTask_particleNet.add( pfParticleNetAK4TaskForRECO, pfParticleNetTask )
-run3_common.toReplaceWith( pfBTaggingTask, _pfBTaggingTask_particleNet)
+_pfBTaggingTask_run3 = cms.Task(
+    # Keep all the infos and DeepCSV
+    pfImpactParameterTagInfos,
+    pfTrackCountingHighEffBJetTags,
+    pfJetProbabilityBJetTags,
+    pfJetBProbabilityBJetTags,
+
+    pfSecondaryVertexTagInfos,
+    inclusiveCandidateVertexingTask,
+    pfInclusiveSecondaryVertexFinderTagInfos,
+    pfGhostTrackVertexTagInfos,
+    pfDeepCSVTask,
+
+    softPFMuonsTagInfos,
+    softPFElectronsTagInfos,
+    pixelClusterTagInfos,
+
+    pfParticleNetAK4TaskForRECO,
+    pfParticleNetTask
+)
+_pfCTaggingTask_run3 = cms.Task(
+    inclusiveCandidateVertexingCvsLTask,
+    pfInclusiveSecondaryVertexFinderCvsLTagInfos,
+)
+run3_common.toReplaceWith( pfBTaggingTask, _pfBTaggingTask_run3 )
+run3_common.toReplaceWith( pfCTaggingTask, _pfCTaggingTask_run3 )