Skip to content

Commit 66577ea

Browse files
authored
Add KM3NeTExtractor to GraphNeTFileReader
1 parent 2343816 commit 66577ea

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

src/graphnet/data/readers/graphnet_file_reader.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
from graphnet.data.extractors.internal import ParquetExtractor
1919
from graphnet.data.extractors.liquido import H5Extractor
2020
from graphnet.data.extractors.prometheus import PrometheusExtractor
21+
from graphnet.data.extractors.km3net import KM3NeTExtractor
2122

2223

2324
class GraphNeTFileReader(Logger, ABC):
@@ -100,6 +101,7 @@ def set_extractors(
100101
List[ParquetExtractor],
101102
List[H5Extractor],
102103
List[PrometheusExtractor],
104+
List[KM3NeTExtractor],
103105
],
104106
) -> None:
105107
"""Set `Extractor`(s) as member variable.
@@ -121,6 +123,7 @@ def _validate_extractors(
121123
List[ParquetExtractor],
122124
List[H5Extractor],
123125
List[PrometheusExtractor],
126+
List[KM3NeTExtractor],
124127
],
125128
) -> None:
126129
for extractor in extractors:

0 commit comments

Comments
 (0)