Skip to content

Commit 45da89d

Browse files
author
ademoor
committed
Using the same addUnifiedParTAK4 in NanoAOD for both UParT v1 and v2
1 parent 570033a commit 45da89d

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

PhysicsTools/NanoAOD/python/jetsAK4_Puppi_cff.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -138,7 +138,7 @@
138138
## - To be used in nanoAOD_customizeCommon() in nano_cff.py
139139
###############################################################
140140
from PhysicsTools.PatAlgos.tools.jetTools import updateJetCollection
141-
def nanoAOD_addDeepInfoAK4(process,addParticleNet,addRobustParTAK4=False,addUnifiedParTAK4=False,addUnifiedParTAK4V1=False):
141+
def nanoAOD_addDeepInfoAK4(process,addParticleNet,addRobustParTAK4=False,addUnifiedParTAK4=False):
142142
_btagDiscriminators=[]
143143
if addParticleNet:
144144
print("Updating process to run ParticleNetAK4")
@@ -154,7 +154,6 @@ def nanoAOD_addDeepInfoAK4(process,addParticleNet,addRobustParTAK4=False,addUnif
154154
print("Updating process to run UnifiedParTAK4")
155155
from RecoBTag.ONNXRuntime.pfUnifiedParticleTransformerAK4_cff import _pfUnifiedParticleTransformerAK4JetTagsAll as pfUnifiedParticleTransformerAK4JetTagsAll
156156
_btagDiscriminators += pfUnifiedParticleTransformerAK4JetTagsAll
157-
if addUnifiedParTAK4V1:
158157
print("Updating process to run UnifiedParTAK4V1")
159158
from RecoBTag.ONNXRuntime.pfUnifiedParticleTransformerAK4V1_cff import _pfUnifiedParticleTransformerAK4V1JetTagsAll as pfUnifiedParticleTransformerAK4V1JetTagsAll
160159
_btagDiscriminators += pfUnifiedParticleTransformerAK4V1JetTagsAll

0 commit comments

Comments
 (0)