diff --git a/boostedhiggs/corrections.py b/boostedhiggs/corrections.py index 340722ced..ca344206b 100644 --- a/boostedhiggs/corrections.py +++ b/boostedhiggs/corrections.py @@ -1010,7 +1010,8 @@ def getGenLepGenQuarks(dataset, genparts: GenParticleArray): elif "TT" in dataset: tops = genparts[get_pid_mask(genparts, TOP_PDGID, byall=False) * genparts.hasFlags(GEN_FLAGS)] - print(tops) + print(tops[:, 0]) + # take all possible daughters! daughters = ak.flatten(tops.distinctChildren, axis=2) daughters = daughters[daughters.hasFlags(GEN_FLAGS)]