From 25ce1183f339a6fb33e3305cb381ad3efc278378 Mon Sep 17 00:00:00 2001 From: "askerosted@gmail.com" Date: Tue, 17 Dec 2024 13:51:01 +0900 Subject: [PATCH] add extractors to init --- src/graphnet/data/extractors/icecube/__init__.py | 5 +++++ .../data/extractors/icecube/i3highesteparticleextractor.py | 2 +- 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/src/graphnet/data/extractors/icecube/__init__.py b/src/graphnet/data/extractors/icecube/__init__.py index 11befe581..40d05be68 100644 --- a/src/graphnet/data/extractors/icecube/__init__.py +++ b/src/graphnet/data/extractors/icecube/__init__.py @@ -18,3 +18,8 @@ from .i3pisaextractor import I3PISAExtractor from .i3ntmuonlabelsextractor import I3NTMuonLabelExtractor from .i3quesoextractor import I3QUESOExtractor +from .i3dictvalueextractor import I3DictValueExtractor +from .i3filtermapextractor import I3FilterMapExtractor +from .i3mclabeler import I3MCLabelerExtractor +from .i3highesteparticleextractor import I3HighestEparticleExtractor +from .i3totaleextractor import I3TotalEExtractor diff --git a/src/graphnet/data/extractors/icecube/i3highesteparticleextractor.py b/src/graphnet/data/extractors/icecube/i3highesteparticleextractor.py index 616bb4fe6..2fa1faf5a 100644 --- a/src/graphnet/data/extractors/icecube/i3highesteparticleextractor.py +++ b/src/graphnet/data/extractors/icecube/i3highesteparticleextractor.py @@ -9,7 +9,7 @@ import numpy as np -class HigestEparticleExtractor(I3Extractor): +class I3HighestEparticleExtractor(I3Extractor): """Extract the highest energy particle in the event.""" def __init__(