fix: circular import with InterpolationMode

ZettaAI · Dec 23, 2024 · 78447e1 · 78447e1
1 parent 5a1031e
commit 78447e1
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/zetta_utils/tensor_ops/__init__.py b/zetta_utils/tensor_ops/__init__.py
@@ -1,6 +1,3 @@
-from . import generators, traceback_supress
-
-from . import common, convert, label, mask, multitensor, normalization
 from .common import (
     InterpolationMode,
     compare,
@@ -15,3 +12,8 @@
 from .convert import astype, to_np, to_torch
 from .label import get_disp_pair, seg_to_aff, seg_to_rgb
 from .mask import filter_cc  # , coarsen
+
+# Circular import otherwise
+from . import generators, traceback_supress
+
+from . import common, convert, label, mask, multitensor, normalization