From 4b99d861e009f790d55ee0d96b56b7c82ddec5e6 Mon Sep 17 00:00:00 2001
From: veegalinova <veegalinova@gmail.com>
Date: Thu, 8 Jun 2023 14:19:54 +0200
Subject: [PATCH] fix test imports

---
 tests/test_export_models.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/tests/test_export_models.py b/tests/test_export_models.py
index 9d94dda..557a941 100644
--- a/tests/test_export_models.py
+++ b/tests/test_export_models.py
@@ -3,8 +3,6 @@
 from zipfile import ZipFile
 
 import pytest
-from bioimageio.core import load_resource_description
-from bioimageio.core.resource_tests import test_model
 
 from n2v.internals.N2V_DataGenerator import N2V_DataGenerator
 from n2v.models import N2VConfig, N2V
@@ -34,6 +32,9 @@ def temp_dir(tmp_path_factory):
 
 @pytest.mark.bioimage_io
 def test_model_build_and_export_2D_RGB(temp_dir):
+    from bioimageio.core import load_resource_description
+    from bioimageio.core.resource_tests import test_model
+
     str_dir = str(temp_dir)
     str_models = str(Path(temp_dir, "models"))
     model_name = "n2v_2D_RGB"
@@ -87,6 +88,9 @@ def test_model_build_and_export_2D_RGB(temp_dir):
 
 @pytest.mark.bioimage_io
 def test_model_build_and_export_2D_SEM(temp_dir):
+    from bioimageio.core import load_resource_description
+    from bioimageio.core.resource_tests import test_model
+
     str_dir = str(temp_dir)
     str_models = str(Path(temp_dir, "models"))
     model_name = "n2v_2D_SEM"