Skip to content
Snippets Groups Projects
Commit 9dd22594 authored by Ruggero Turra's avatar Ruggero Turra :headphones: Committed by Vakhtang Tsulaia
Browse files

Python3 + flake8 ElectronPhotonFourMomentumCorrection

parent 4cda231e
No related branches found
No related tags found
No related merge requests found
......@@ -66,5 +66,5 @@ if( XAOD_STANDALONE )
endif()
# Install files from the package:
atlas_install_python_modules( python/*.py )
atlas_install_python_modules( python/*.py POST_BUILD_CMD ${ATLAS_FLAKE8} )
atlas_install_joboptions( share/*.py )
......@@ -51,7 +51,7 @@ def merge_histograms(old, new, merge_error=True):
new_values.append(old.GetBinContent(iold))
new_errors.append(old.GetBinError(iold))
if ir_new == OVERFLOW:
remainer = iold
remainer = iold # noqa: F841
print("breaking")
break
last_old = iold
......
......@@ -37,12 +37,12 @@ def xAOD_particle_generator(tree, collection_getter, newevent_function=None, end
elif type(event_numbers) is int:
event_numbers = [event_numbers]
for ievent in xrange(tree.GetEntries()):
for ievent in range(tree.GetEntries()):
tree.GetEntry(ievent)
ei = tree.EventInfo
event_number = ei.eventNumber()
if event_numbers:
if not event_number in event_numbers:
if event_number not in event_numbers:
continue
logging.debug("=== event number %d ievent = %d", event_number, ievent)
if newevent_function is not None:
......@@ -50,7 +50,7 @@ def xAOD_particle_generator(tree, collection_getter, newevent_function=None, end
collection = collection_getter(tree)
for i in xrange(collection.size()):
for i in range(collection.size()):
p = collection.at(i)
if min_pt is not None and p.pt() < min_pt:
continue
......@@ -93,7 +93,7 @@ def main(filename, **args):
f.Print()
return
logging.info("input has %d entries" % tree.GetEntries())
logging.info("input has %d entries", tree.GetEntries())
logging.debug("initializing tool")
tool = ROOT.CP.EgammaCalibrationAndSmearingTool("tool")
......@@ -168,7 +168,6 @@ def main(filename, **args):
fout.Close()
if __name__ == '__main__':
ROOT.gROOT.ProcessLine(".x $ROOTCOREDIR/scripts/load_packages.C")
import argparse
parser = argparse.ArgumentParser(description='Run on xAOD and dump calibrated energy for electron and photons',
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment