-
Notifications
You must be signed in to change notification settings - Fork 94
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Improvements for inelasticity reconstructions and other utilities #763
base: main
Are you sure you want to change the base?
Conversation
@@ -13,7 +13,7 @@ | |||
from graphnet.utilities.imports import has_icecube_package | |||
|
|||
if has_icecube_package() or TYPE_CHECKING: | |||
from icecube import ( | |||
from icecube import ( # noqa: F401 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I needed to add this because LeptonInjector isn't used directly, but it needs to be imported for icetray to recognize the "EventProperties" frame object type.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@pweigel thank you for these contributions!
I added a small comment related to the duplication of the set_gcd
.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
🚀
This pull request implements some changes for training inelasticity reconstructions and adds other utilities:
I3TruthExtractor
can now determine starting events using the event vertex and the convex hull of the detector volume (using the GCD file). The convex hull can also be expanded withextend_boundary
. Some other utilities in this class use the "boundaries" list, but I think these could be redone to use the convex hull. I should note I had to override theset_gcd
function since the geometry is added after instantiation.I3TruthExtractor
can now identify MC produced with LeptonInjector and get the relevant quantities.sub_event_stream
.IceMixNodes
to enable/disable the random pulse sampling (default True).VisibleInelasticityReconstruction
taskMAELoss