Add keras test model

Author: constantinpape

tests/build_spec/test_build_spec.py

@@ -67,7 +67,6 @@ def _test_build_spec(
         output_path=out_path,
         add_deepimagej_config=add_deepimagej_config,
     )
-    # TODO names
     if architecture is not None:
         kwargs["architecture"] = architecture
     if model_kwargs is not None:
@@ -134,5 +133,5 @@ def test_build_spec_deepimagej(unet2d_nuclei_broad_model, tmp_path):
     _test_build_spec(unet2d_nuclei_broad_model, tmp_path / "model.zip", "pytorch_script", add_deepimagej_config=True)
 
 
-# def test_build_spec_deepimagej_keras(unet2d_keras, tmp_path):
-#     _test_build_spec(unet2d_keras, tmp_path / "model.zip", "pytorch_script", add_deepimagej_config=True)
+def test_build_spec_deepimagej_keras(unet2d_keras, tmp_path):
+    _test_build_spec(unet2d_keras, tmp_path / "model.zip", "keras_hdf5", add_deepimagej_config=True)

tests/conftest.py

@@ -13,12 +13,14 @@
 onnx_models = ["unet2d_multi_tensor", "unet2d_nuclei_broad_model", "hpa_densenet"]
 tensorflow1_models = ["stardist"]
 tensorflow2_models = []
-keras_models = []
+keras_models = ["unet2d_keras"]
 tensorflow_js_models = []
 
 model_sources = {
-    # TODO add unet2d_keras_tf from https://github.com/bioimage-io/spec-bioimage-io/pull/267
-    # "unet2d_keras_tf": (""),
+    "unet2d_keras": (
+        "https://raw.githubusercontent.com/bioimage-io/spec-bioimage-io/main/example_specs/models/"
+        "unet2d_keras_tf/rdf.yaml"
+    ),
     "unet2d_nuclei_broad_model": (
         "https://raw.githubusercontent.com/bioimage-io/spec-bioimage-io/main/example_specs/models/"
         "unet2d_nuclei_broad/rdf.yaml"
@@ -79,7 +81,6 @@
 except ImportError:
     keras = None
 skip_keras = keras is None
-skip_keras = True  # TODO add unet2d_keras_tf to have a model for keras tests
 
 # load all model packages we need for testing
 load_model_packages = set()
@@ -197,3 +198,10 @@ def unet2d_fixed_shape_or_not(request):
 )
 def unet2d_multi_tensor_or_not(request):
     return pytest.model_packages[request.param]
+
+
+@pytest.fixture(
+    params=[] if skip_keras else ["unet2d_keras"]
+)
+def unet2d_keras(request):
+    return pytest.model_packages[request.param]