small change missed from new specs

bioimage-io · Nov 12, 2024 · 7070a45 · 7070a45
1 parent 6767969
commit 7070a45
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/ModelWeight.java b/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/ModelWeight.java
@@ -78,6 +78,8 @@ public static ModelWeight build(Map<String, Object> yamlFieldElements)
         model.weightsDic = new HashMap<String, WeightFormat>();
         for (String ww : weightsFormats) {
         	Map<String, Object> weights = (Map<String, Object>) yamlFieldElements.get(ww);
+        	if (weights == null)
+        		continue;
 	        if (ww.contentEquals(kerasIdentifier)) {
 	        	KerasWeights weightsObject = new KerasWeights(weights);
 	        	model.weightsDic.put(model.kerasEngineName(weightsObject), weightsObject);