move import into function

juglab · Jun 8, 2023 · 1c09ca6 · 1c09ca6
1 parent 4b99d86
commit 1c09ca6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/n2v/utils/export_utils.py b/n2v/utils/export_utils.py
@@ -5,7 +5,6 @@
 import tensorflow as tf
 from zipfile import ZipFile
 from csbdeep.utils import save_json
-from bioimageio.core.build_spec import build_model
 
 from ..models.n2v_config import N2VConfig
 
@@ -124,6 +123,8 @@ def build_modelzoo(
     files: list = [],
     **kwargs,
 ):
+    from bioimageio.core.build_spec import build_model
+
     tags_dim = "3d" if len(axes) == 5 else "2d"
 
     build_model(