improe robustness

bioimage-io · Jan 9, 2025 · efe0f0b · efe0f0b
1 parent a424ba1
commit efe0f0b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main/java/io/bioimage/modelrunner/bioimageio/download/DownloadTracker.java b/src/main/java/io/bioimage/modelrunner/bioimageio/download/DownloadTracker.java
@@ -500,7 +500,8 @@ public static void printProgress(Thread downloadThread,
 		Set<String> already = new HashSet<String>();
 		while (Thread.currentThread().isAlive() && downloadThread.isAlive()) {
 			long waitMillis = 0;
-			while (consumer.get().get(TOTAL_PROGRESS_KEY) < 1.0 && waitMillis < 3000) {
+			while ((consumer.get().get(TOTAL_PROGRESS_KEY) == null || consumer.get().get(TOTAL_PROGRESS_KEY) < 1.0)
+					&& waitMillis < 3000) {
 				Thread.sleep(30);
 				waitMillis += 30;
 			}