Merge pull request graphnet-team#772 from pweigel/fix_i3tray

Fix deprecated I3Tray imports

Author: pweigel

docker/gnn-benchmarking/apply.py

@@ -6,7 +6,7 @@
 from os.path import join, dirname
 from typing import List
 
-from I3Tray import I3Tray  # pyright: reportMissingImports=false
+from icecube.icetray import I3Tray  # pyright: reportMissingImports=false
 
 from graphnet.deployment.i3modules import I3InferenceModule
 from graphnet.data.extractors.i3featureextractor import (

examples/01_icetray/04_i3_module_in_native_icetray_example.py

@@ -19,7 +19,7 @@
 
 if has_icecube_package() or TYPE_CHECKING:
     from icecube import icetray, dataio  # pyright: reportMissingImports=false
-    from I3Tray import I3Tray
+    from icecube.icetray import I3Tray
 
     from graphnet.deployment.i3modules import (
         I3InferenceModule,

src/graphnet/deployment/icecube/i3deployer.py

@@ -11,7 +11,7 @@
 
 if has_icecube_package() or TYPE_CHECKING:
     from icecube import icetray, dataio  # pyright: reportMissingImports=false
-    from I3Tray import I3Tray
+    from icecube.icetray import I3Tray
 
 
 class I3Deployer(Deployer):

tests/deployment/queso_test.py

@@ -19,7 +19,7 @@
 
 if has_icecube_package() or TYPE_CHECKING:
     from icecube import icetray, dataio  # pyright: reportMissingImports=false
-    from I3Tray import I3Tray
+    from icecube.icetray import I3Tray
 
     from graphnet.deployment.i3modules import (
         I3InferenceModule,