Skip to content

Commit fe15700

Browse files
committed
move test_model imports
1 parent 5563e92 commit fe15700

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/test_weight_converters.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,12 +5,12 @@
55

66
import pytest
77

8-
from bioimageio.core import test_model
98
from bioimageio.spec import load_description
109
from bioimageio.spec.model import v0_5
1110

1211

1312
def test_pytorch_to_torchscript(any_torch_model, tmp_path):
13+
from bioimageio.core import test_model
1414
from bioimageio.core.weight_converters.pytorch_to_torchscript import convert
1515

1616
model_descr = load_description(any_torch_model, perform_io_checks=False)
@@ -28,6 +28,7 @@ def test_pytorch_to_torchscript(any_torch_model, tmp_path):
2828

2929

3030
def test_pytorch_to_onnx(convert_to_onnx, tmp_path):
31+
from bioimageio.core import test_model
3132
from bioimageio.core.weight_converters.pytorch_to_onnx import convert
3233

3334
model_descr = load_description(convert_to_onnx, format_version="latest")
@@ -50,6 +51,7 @@ def test_pytorch_to_onnx(convert_to_onnx, tmp_path):
5051

5152
@pytest.mark.skip()
5253
def test_keras_to_tensorflow(any_keras_model: Path, tmp_path: Path):
54+
from bioimageio.core import test_model
5355
from bioimageio.core.weight_converters.keras_to_tensorflow import convert
5456

5557
out_path = tmp_path / "weights.zip"

0 commit comments

Comments
 (0)