Skip to content

Commit 80bf0e4

Browse files
committed
don't test onnx converter with 'any_torch_model' yet
1 parent e3312e0 commit 80bf0e4

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/weight_converter/torch/test_onnx.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,11 +2,12 @@
22
import pytest
33

44

5-
def test_onnx_converter_from_torch(any_torch_model, tmp_path):
5+
# todo: test with 'any_torch_model'
6+
def test_onnx_converter_from_torch(unet2d_nuclei_broad, tmp_path):
67
from bioimageio.core.weight_converter.torch.onnx import convert_weights_to_onnx
78

89
out_path = tmp_path / "weights.onnx"
9-
ret_val = convert_weights_to_onnx(any_torch_model, out_path, test_decimal=3)
10+
ret_val = convert_weights_to_onnx(unet2d_nuclei_broad, out_path, test_decimal=3)
1011
assert os.path.exists(out_path)
1112
if not pytest.skip_onnx:
1213
assert ret_val == 0 # check for correctness is done in converter and returns 0 if it passes

0 commit comments

Comments
 (0)