From 78447e195289b8d74133b901d94db6dd8f64ac33 Mon Sep 17 00:00:00 2001
From: Nico Kemnitz <nico@zetta.ai>
Date: Mon, 23 Dec 2024 13:46:03 +0100
Subject: [PATCH] fix: circular import with InterpolationMode

---
 zetta_utils/tensor_ops/__init__.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/zetta_utils/tensor_ops/__init__.py b/zetta_utils/tensor_ops/__init__.py
index 9b05678e3..44d1880ec 100644
--- 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