From 53935b6d3d14d18ccbbe26649e59fcb55d6692b7 Mon Sep 17 00:00:00 2001
From: Tianlu Yuan <5412915+tianluyuan@users.noreply.github.com>
Date: Wed, 13 Sep 2023 21:35:07 -0500
Subject: [PATCH] update deprecated from I3Tray

---
 skymap_scanner/client/reco_icetray.py      | 2 +-
 skymap_scanner/recos/dummy.py              | 2 +-
 skymap_scanner/recos/millipede_original.py | 2 +-
 skymap_scanner/recos/millipede_wilks.py    | 2 +-
 skymap_scanner/recos/splinempe.py          | 2 +-
 skymap_scanner/server/start_scan.py        | 2 +-
 skymap_scanner/utils/prepare_frames.py     | 2 +-
 7 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/skymap_scanner/client/reco_icetray.py b/skymap_scanner/client/reco_icetray.py
index b2ef4f0c8..60048f22c 100644
--- a/skymap_scanner/client/reco_icetray.py
+++ b/skymap_scanner/client/reco_icetray.py
@@ -11,7 +11,7 @@
 from pathlib import Path
 from typing import Any, List, Union
 
-from I3Tray import I3Tray  # type: ignore[import]
+from icecube.icetray import I3Tray  # type: ignore[import]
 from icecube import (  # type: ignore[import]  # noqa: F401
     dataio,
     frame_object_diff,
diff --git a/skymap_scanner/recos/dummy.py b/skymap_scanner/recos/dummy.py
index 2c19bc7ba..a753ffc09 100644
--- a/skymap_scanner/recos/dummy.py
+++ b/skymap_scanner/recos/dummy.py
@@ -6,7 +6,7 @@
 import time
 from typing import List, Final
 
-from I3Tray import I3Units  # type: ignore[import]
+from icecube.icetray import I3Units  # type: ignore[import]
 from icecube import (  # type: ignore[import]  # noqa: F401
     dataclasses,
     frame_object_diff,
diff --git a/skymap_scanner/recos/millipede_original.py b/skymap_scanner/recos/millipede_original.py
index 7ca905395..2ef5cb34f 100644
--- a/skymap_scanner/recos/millipede_original.py
+++ b/skymap_scanner/recos/millipede_original.py
@@ -10,7 +10,7 @@
 
 import numpy
 
-from I3Tray import I3Units
+from icecube.icetray import I3Units
 from icecube import (  # noqa: F401
     VHESelfVeto,
     dataclasses,
diff --git a/skymap_scanner/recos/millipede_wilks.py b/skymap_scanner/recos/millipede_wilks.py
index 26d6f9b36..ee3a970fd 100644
--- a/skymap_scanner/recos/millipede_wilks.py
+++ b/skymap_scanner/recos/millipede_wilks.py
@@ -10,7 +10,7 @@
 from typing import Final, List, Tuple
 
 import numpy
-from I3Tray import I3Units
+from icecube.icetray import I3Units
 from icecube import (  # noqa: F401
     VHESelfVeto,
     dataclasses,
diff --git a/skymap_scanner/recos/splinempe.py b/skymap_scanner/recos/splinempe.py
index 92a781fb6..96fda512c 100644
--- a/skymap_scanner/recos/splinempe.py
+++ b/skymap_scanner/recos/splinempe.py
@@ -6,7 +6,7 @@
 from typing import Final, List
 
 
-from I3Tray import I3Units  # type: ignore[import]
+from icecube.icetray import I3Units  # type: ignore[import]
 
 # NOTE: icecube module imports are required to make IceTray modules and services available.
 from icecube import (  # type: ignore[import]  # noqa: F401
diff --git a/skymap_scanner/server/start_scan.py b/skymap_scanner/server/start_scan.py
index c99659c7a..67aff6278 100644
--- a/skymap_scanner/server/start_scan.py
+++ b/skymap_scanner/server/start_scan.py
@@ -15,7 +15,7 @@
 import healpy  # type: ignore[import]
 import mqclient as mq
 import numpy
-from I3Tray import I3Units  # type: ignore[import]
+from icecube.icetray import I3Units  # type: ignore[import]
 from icecube import (  # type: ignore[import]
     astro,
     dataclasses,
diff --git a/skymap_scanner/utils/prepare_frames.py b/skymap_scanner/utils/prepare_frames.py
index 2545a3f26..f7155652a 100644
--- a/skymap_scanner/utils/prepare_frames.py
+++ b/skymap_scanner/utils/prepare_frames.py
@@ -9,7 +9,7 @@
 import os
 from typing import Union, List
 
-from I3Tray import I3Tray  # type: ignore[import]
+from icecube.icetray import I3Tray  # type: ignore[import]
 from icecube import icetray  # type: ignore[import]
 from icecube.frame_object_diff.segments import uncompress  # type: ignore[import]