add more information to errors

bioimage-io · Dec 20, 2024 · 979f446 · 979f446
1 parent 84c98d2
commit 979f446
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/main/java/io/bioimage/modelrunner/tensorflow/v1/Tensorflow1Interface.java b/src/main/java/io/bioimage/modelrunner/tensorflow/v1/Tensorflow1Interface.java
@@ -220,11 +220,11 @@ private void launchModelLoadOnProcess() throws IOException, InterruptedException
 		if (task.status == TaskStatus.CANCELED)
 			throw new RuntimeException();
 		else if (task.status == TaskStatus.FAILED)
-			throw new RuntimeException();
+			throw new RuntimeException(task.error);
 		else if (task.status == TaskStatus.CRASHED) {
 			this.runner.close();
 			runner = null;
-			throw new RuntimeException();
+			throw new RuntimeException(task.error);
 		}
 	}
 
@@ -380,11 +380,11 @@ void runInterprocessing(List<Tensor<T>> inputTensors, List<Tensor<R>> outputTens
 			if (task.status == TaskStatus.CANCELED)
 				throw new RuntimeException();
 			else if (task.status == TaskStatus.FAILED)
-				throw new RuntimeException();
+				throw new RuntimeException(task.error);
 			else if (task.status == TaskStatus.CRASHED) {
 				this.runner.close();
 				runner = null;
-				throw new RuntimeException();
+				throw new RuntimeException(task.error);
 			}
 			for (int i = 0; i < outputTensors.size(); i ++) {
 	        	String name = (String) Types.decode(encOuts.get(i)).get(MEM_NAME_KEY);
@@ -508,11 +508,11 @@ public void closeModel() {
 			if (task.status == TaskStatus.CANCELED)
 				throw new RuntimeException();
 			else if (task.status == TaskStatus.FAILED)
-				throw new RuntimeException();
+				throw new RuntimeException(task.error);
 			else if (task.status == TaskStatus.CRASHED) {
 				this.runner.close();
 				runner = null;
-				throw new RuntimeException();
+				throw new RuntimeException(task.error);
 			}
 			this.runner.close();
 			this.runner = null;