dataclasses missing fix

Aske-Rosted · Dec 18, 2024 · 79dc6bc · 79dc6bc
1 parent 0baa572
commit 79dc6bc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/graphnet/data/extractors/icecube/i3highesteparticleextractor.py b/src/graphnet/data/extractors/icecube/i3highesteparticleextractor.py
@@ -2,7 +2,7 @@
 
 from typing import Dict, Any, List, TYPE_CHECKING
 
-from .utilities.gcd_hull import GCD_hull
+
 from .i3extractor import I3Extractor
 
 import numpy as np
@@ -15,6 +15,7 @@
         dataclasses,
         MuonGun,
     )  # pyright: reportMissingImports=false
+    from .utilities.gcd_hull import GCD_hull
 
 
 class I3HighestEparticleExtractor(I3Extractor):

diff --git a/src/graphnet/data/extractors/icecube/i3totaleextractor.py b/src/graphnet/data/extractors/icecube/i3totaleextractor.py
@@ -2,7 +2,7 @@
 
 from typing import Dict, Any, List, TYPE_CHECKING
 
-from .utilities.gcd_hull import GCD_hull
+
 from .i3extractor import I3Extractor
 
 import numpy as np
@@ -15,6 +15,7 @@
         dataclasses,
         MuonGun,
     )  # pyright: reportMissingImports=false
+    from .utilities.gcd_hull import GCD_hull
 
 
 class I3TotalEExtractor(I3Extractor):