Fix small issues in export scripts

constantinpape · Jul 15, 2021 · 69cf080 · 69cf080
1 parent 089ad2a
commit 69cf080
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/experiments/neuron-segmentation/cremi/export_bioimageio_model.py b/experiments/neuron-segmentation/cremi/export_bioimageio_model.py
@@ -88,6 +88,7 @@ def export_to_bioimageio(checkpoint, output, input_, affs_to_bd, additional_form
         cite=cite,
         model_postprocessing=postprocessing,
         input_optional_parameters=False,
+        for_deepimagej="torchscript" in additional_formats,
         links=[get_bioimageio_dataset_id("cremi")]
     )
     add_weight_formats(output, additional_formats)

diff --git a/experiments/platynereis/export_bioimageio_model.py b/experiments/platynereis/export_bioimageio_model.py
@@ -95,6 +95,7 @@ def export_to_bioimageio(checkpoint, output, input_, affs_to_bd, additional_form
         cite=cite,
         model_postprocessing=postprocessing,
         input_optional_parameters=False,
+        for_deepimagej="torchscript" in additional_formats,
         links=[get_bioimageio_dataset_id("platynereis")]
     )
     add_weight_formats(output, additional_formats)