From 762c4ce9a4728823ceebfc7287cb100efd2bc379 Mon Sep 17 00:00:00 2001 From: carlosuc3m <100329787@alumnos.uc3m.es> Date: Fri, 20 Dec 2024 13:09:42 +0100 Subject: [PATCH] add more information to errors --- .../modelrunner/pytorch/PytorchInterface.java | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/src/main/java/io/bioimage/modelrunner/pytorch/PytorchInterface.java b/src/main/java/io/bioimage/modelrunner/pytorch/PytorchInterface.java index 16f6826..b6290ea 100644 --- a/src/main/java/io/bioimage/modelrunner/pytorch/PytorchInterface.java +++ b/src/main/java/io/bioimage/modelrunner/pytorch/PytorchInterface.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); } } @@ -323,11 +323,11 @@ void runInterprocessing(List> inputTensors, List> 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); @@ -451,11 +451,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;