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 ) ...@@ -66,5 +66,5 @@ if( XAOD_STANDALONE )
endif() endif()
# Install files from the package: # 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 ) atlas_install_joboptions( share/*.py )
...@@ -51,7 +51,7 @@ def merge_histograms(old, new, merge_error=True): ...@@ -51,7 +51,7 @@ def merge_histograms(old, new, merge_error=True):
new_values.append(old.GetBinContent(iold)) new_values.append(old.GetBinContent(iold))
new_errors.append(old.GetBinError(iold)) new_errors.append(old.GetBinError(iold))
if ir_new == OVERFLOW: if ir_new == OVERFLOW:
remainer = iold remainer = iold # noqa: F841
print("breaking") print("breaking")
break break
last_old = iold last_old = iold
......
...@@ -37,12 +37,12 @@ def xAOD_particle_generator(tree, collection_getter, newevent_function=None, end ...@@ -37,12 +37,12 @@ def xAOD_particle_generator(tree, collection_getter, newevent_function=None, end
elif type(event_numbers) is int: elif type(event_numbers) is int:
event_numbers = [event_numbers] event_numbers = [event_numbers]
for ievent in xrange(tree.GetEntries()): for ievent in range(tree.GetEntries()):
tree.GetEntry(ievent) tree.GetEntry(ievent)
ei = tree.EventInfo ei = tree.EventInfo
event_number = ei.eventNumber() event_number = ei.eventNumber()
if event_numbers: if event_numbers:
if not event_number in event_numbers: if event_number not in event_numbers:
continue continue
logging.debug("=== event number %d ievent = %d", event_number, ievent) logging.debug("=== event number %d ievent = %d", event_number, ievent)
if newevent_function is not None: if newevent_function is not None:
...@@ -50,7 +50,7 @@ def xAOD_particle_generator(tree, collection_getter, newevent_function=None, end ...@@ -50,7 +50,7 @@ def xAOD_particle_generator(tree, collection_getter, newevent_function=None, end
collection = collection_getter(tree) collection = collection_getter(tree)
for i in xrange(collection.size()): for i in range(collection.size()):
p = collection.at(i) p = collection.at(i)
if min_pt is not None and p.pt() < min_pt: if min_pt is not None and p.pt() < min_pt:
continue continue
...@@ -93,7 +93,7 @@ def main(filename, **args): ...@@ -93,7 +93,7 @@ def main(filename, **args):
f.Print() f.Print()
return return
logging.info("input has %d entries" % tree.GetEntries()) logging.info("input has %d entries", tree.GetEntries())
logging.debug("initializing tool") logging.debug("initializing tool")
tool = ROOT.CP.EgammaCalibrationAndSmearingTool("tool") tool = ROOT.CP.EgammaCalibrationAndSmearingTool("tool")
...@@ -168,7 +168,6 @@ def main(filename, **args): ...@@ -168,7 +168,6 @@ def main(filename, **args):
fout.Close() fout.Close()
if __name__ == '__main__': if __name__ == '__main__':
ROOT.gROOT.ProcessLine(".x $ROOTCOREDIR/scripts/load_packages.C")
import argparse import argparse
parser = argparse.ArgumentParser(description='Run on xAOD and dump calibrated energy for electron and photons', 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