Merge pull request #174 from susy2015/typeCast

Explicitly typecast ints to fix issue between numpy ints and TTreeRe…
susy2015 · Apr 12, 2019 · ca8dfa1 · ca8dfa1
2 parents fef7576 + 1ec08b5
commit ca8dfa1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/TopTagger/python/TopTaggerProducer.py b/TopTagger/python/TopTaggerProducer.py
@@ -162,9 +162,9 @@ def analyze(self, event):
         self.out.fillBranch("ResolvedTop%s_phi"%self.suffixResolved,           ttr.phiCol()[resolvedFilter])
         self.out.fillBranch("ResolvedTop%s_mass"%self.suffixResolved,          ttr.massCol()[resolvedFilter])
         self.out.fillBranch("ResolvedTop%s_discriminator"%self.suffixResolved, ttr.discCol()[resolvedFilter])
-        self.out.fillBranch("ResolvedTop%s_j1Idx"%self.suffixResolved,         ttr.j1IdxCol()[resolvedFilter])
-        self.out.fillBranch("ResolvedTop%s_j2Idx"%self.suffixResolved,         ttr.j2IdxCol()[resolvedFilter])
-        self.out.fillBranch("ResolvedTop%s_j3Idx"%self.suffixResolved,         ttr.j3IdxCol()[resolvedFilter])
+        self.out.fillBranch("ResolvedTop%s_j1Idx"%self.suffixResolved,         ttr.j1IdxCol()[resolvedFilter].astype(int))
+        self.out.fillBranch("ResolvedTop%s_j2Idx"%self.suffixResolved,         ttr.j2IdxCol()[resolvedFilter].astype(int))
+        self.out.fillBranch("ResolvedTop%s_j3Idx"%self.suffixResolved,         ttr.j3IdxCol()[resolvedFilter].astype(int))
 
         if self.saveAK8:
             self.out.fillBranch("MergedTop%s_pt"%self.suffix,              ttr.ptCol()[mergedFilter])