keep increasing robustness

bioimage-io · Oct 9, 2024 · c81da68 · c81da68
1 parent f6bd476
commit c81da68
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/main/java/io/bioimage/modelrunner/bioimageio/description/TensorSpecV05.java b/src/main/java/io/bioimage/modelrunner/bioimageio/description/TensorSpecV05.java
@@ -73,7 +73,10 @@ public class TensorSpecV05 implements TensorSpec {
      */
     protected TensorSpecV05(Map<String, Object> tensorSpecMap, boolean input)
     {
-        id = (String) tensorSpecMap.get("name");
+        if (tensorSpecMap.get("id") == null)
+        	id = (String) tensorSpecMap.get("name");
+        else
+        	id = (String) tensorSpecMap.get("id");
         if (tensorSpecMap.get("axes") == null || !(tensorSpecMap.get("axes") instanceof List))
         	throw new IllegalArgumentException("Invalid tensor specifications for '" + id
         			+ "'. The axes are incorrectly specified. For more info, visit the Bioimage.io docs.");