Merge pull request #253 from bioimage-io/fix-torch-to-onnx

constantinpape · web-flow · commit b7f48138906a · 2022-03-20T16:12:39.000+01:00
Fix torch to onnx conversion
diff --git a/bioimageio/core/weight_converter/torch/onnx.py b/bioimageio/core/weight_converter/torch/onnx.py
@@ -58,7 +58,6 @@ def convert_weights_to_onnx(
                 output_path,
                 verbose=verbose,
                 opset_version=opset_version,
-                example_outputs=expected_outputs,
             )
         else:
             raise NotImplementedError
diff --git a/dev/environment-torch.yaml b/dev/environment-torch.yaml
@@ -6,12 +6,12 @@ dependencies:
   - black
   - bioimageio.spec >=0.4.4
   - conda-build
-  - h5py >=2.10,<2.11
+  - h5py
   - mypy
   - pip
   - pytest
   - python >=3.7
   - xarray
-  - pytorch <1.10
+  - pytorch
   - onnxruntime
   - tifffile

Original file line number	Diff line number	Diff line change
`@@ -58,7 +58,6 @@ def convert_weights_to_onnx(`
`58`	`58`	`output_path,`
`59`	`59`	`verbose=verbose,`
`60`	`60`	`opset_version=opset_version,`
`61`		`- example_outputs=expected_outputs,`
`62`	`61`	`)`
`63`	`62`	`else:`
`64`	`63`	`raise NotImplementedError`