We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e3312e0 commit 80bf0e4Copy full SHA for 80bf0e4
tests/weight_converter/torch/test_onnx.py
@@ -2,11 +2,12 @@
2
import pytest
3
4
5
-def test_onnx_converter_from_torch(any_torch_model, tmp_path):
+# todo: test with 'any_torch_model'
6
+def test_onnx_converter_from_torch(unet2d_nuclei_broad, tmp_path):
7
from bioimageio.core.weight_converter.torch.onnx import convert_weights_to_onnx
8
9
out_path = tmp_path / "weights.onnx"
- 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)
11
assert os.path.exists(out_path)
12
if not pytest.skip_onnx:
13
assert ret_val == 0 # check for correctness is done in converter and returns 0 if it passes
0 commit comments