Skip to content

Commit 2585fc7

Browse files
Merge pull request #113 from bioimage-io/add-hpa
Add hpa model to onnx models
2 parents 2ba8ead + c7d7676 commit 2585fc7

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/conftest.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
# test models for various frameworks
55
torch_models = ["unet2d_fixed_shape", "unet2d_multi_tensor", "unet2d_nuclei_broad_model"]
66
torchscript_models = ["unet2d_multi_tensor", "unet2d_nuclei_broad_model"]
7-
onnx_models = ["unet2d_multi_tensor", "unet2d_nuclei_broad_model"]
7+
onnx_models = ["unet2d_multi_tensor", "unet2d_nuclei_broad_model", "hpa_densenet"]
88
tensorflow1_models = ["FruNet_model"]
99
tensorflow2_models = []
1010
keras_models = ["FruNet_model"]
@@ -25,6 +25,9 @@
2525
"https://raw.githubusercontent.com/bioimage-io/spec-bioimage-io/main/example_specs/models/"
2626
"unet2d_multi_tensor/rdf.yaml"
2727
),
28+
"hpa_densenet": (
29+
"https://raw.githubusercontent.com/bioimage-io/spec-bioimage-io/main/example_specs/models/hpa-densenet/rdf.yaml"
30+
),
2831
}
2932

3033
try:

0 commit comments

Comments
 (0)