Merge pull request #113 from bioimage-io/add-hpa

Add hpa model to onnx models
bioimage-io · Sep 24, 2021 · 2585fc7 · 2585fc7
2 parents 2ba8ead + c7d7676
commit 2585fc7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -4,7 +4,7 @@
 # test models for various frameworks
 torch_models = ["unet2d_fixed_shape", "unet2d_multi_tensor", "unet2d_nuclei_broad_model"]
 torchscript_models = ["unet2d_multi_tensor", "unet2d_nuclei_broad_model"]
-onnx_models = ["unet2d_multi_tensor", "unet2d_nuclei_broad_model"]
+onnx_models = ["unet2d_multi_tensor", "unet2d_nuclei_broad_model", "hpa_densenet"]
 tensorflow1_models = ["FruNet_model"]
 tensorflow2_models = []
 keras_models = ["FruNet_model"]
@@ -25,6 +25,9 @@
         "https://raw.githubusercontent.com/bioimage-io/spec-bioimage-io/main/example_specs/models/"
         "unet2d_multi_tensor/rdf.yaml"
     ),
+    "hpa_densenet": (
+        "https://raw.githubusercontent.com/bioimage-io/spec-bioimage-io/main/example_specs/models/hpa-densenet/rdf.yaml"
+    ),
 }
 
 try: