add more information to errors

bioimage-io · Dec 20, 2024 · 4818559 · 4818559
1 parent cb2a417
commit 4818559
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/main/java/io/bioimage/modelrunner/tensorflow/v2/api050/Tensorflow2Interface.java b/src/main/java/io/bioimage/modelrunner/tensorflow/v2/api050/Tensorflow2Interface.java
@@ -198,11 +198,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);
 		}
 	}
 
@@ -363,11 +363,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);
@@ -487,11 +487,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;