Fix I3HighestEparticleExtractor #822
Open
+137
−39
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This PR addresses the bugs explained in #821.
Solving Issues
1. Negative visible length
set visible length to 0 if it would be negative
2. Track filtering only for top-level primaries
Look, if a track particle is in the subtree of any primary, not if the head-level primary is in the
primaries
list3. Wrong return in
get_bundle_HEP
return the right tuple
Additionally, I introduced the
get_descendants
function, which accelerates this code snippet:graphnet/src/graphnet/data/extractors/icecube/i3highesteparticleextractor.py
Lines 399 to 414 in f3faff1
by not iterating through all particles, but only the descendants of the primaries