always unzip weigths when downloading

bioimage-io · Nov 2, 2023 · ed4c088 · ed4c088
1 parent 285098e
commit ed4c088
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/java/io/bioimage/modelrunner/bioimageio/download/DownloadModel.java b/src/main/java/io/bioimage/modelrunner/bioimageio/download/DownloadModel.java
@@ -269,9 +269,9 @@ private void addWeights() {
 				WeightFormat w = weights.getWeightsByIdentifier(ww);
 				if (w.getSource() != null && checkURL(w.getSource())) {
 					downloadableLinks.put(WEIGHTS_KEY + "_" + c ++, w.getSource());
+					if (w.getSourceFileName().endsWith(".zip"))
+						unzip = true;
 				}
-				if (downloadableLinks.get(WEIGHTS_KEY + "_" + (c - 1)).endsWith(".zip"))
-					unzip = true;
 			} catch (Exception ex) {
 				// The exception is thrown whenever the weight format is not present.
 				// This exception will not be thrown here because the weight formats are retrieved from the same object