Protect against missing event branch

susy2015 · Feb 14, 2020 · ef63b41 · ef63b41
1 parent 6a1a28e
commit ef63b41
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/TopTagger/python/TopTagger.py b/TopTagger/python/TopTagger.py
@@ -273,7 +273,11 @@ def getTopsFromExampleFile(tt, event):
 
         event.GetEntry(iEvt)
 
-        print "Event #:", event.event
+        try:
+            evtNum = getattr(event, options.evtBranch)
+            print "Event #:", evtNum
+        except AttributeError:
+            print "Event #:", iEvt
 
         if options.example:
             tops = getTopsFromExampleFile(tt, event)