try more prints

bioimage-io · Nov 23, 2024 · fc1cfd2 · fc1cfd2
1 parent 18b81ef
commit fc1cfd2
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/main/java/io/bioimage/modelrunner/pytorch/javacpp/JavaWorker.java b/src/main/java/io/bioimage/modelrunner/pytorch/javacpp/JavaWorker.java
@@ -35,6 +35,7 @@ public static void main(String[] args) {
     		try {
 				pi = new PytorchJavaCPPInterface(false);
 			} catch (IOException | URISyntaxException e) {
+				e.printStackTrace();
 				return;
 			}
 
@@ -87,13 +88,16 @@ private void executeScript(String script, Map<String, Object> inputs) {
 		this.reportLaunch();
 		try {
 			if (script.equals("loadModel")) {
-				update("STATY IN WORKER LOAD", null, null);
+				update("STATY IN WORKER LOAD LOAD", null, null);
 				pi.loadModel((String) inputs.get("modelFolder"), (String) inputs.get("modelSource"));
 			} else if (script.equals("inference")) {
-				update("STATY IN WORKER", null, null);
+				update("STATY IN WORKER ------------RUN", null, null);
 				pi.runFromShmas((List<String>) inputs.get("inputs"), (List<String>) inputs.get("outputs"));
 			} else if (script.equals("close")) {
 				pi.closeModel();
+			} else {
+				update("LOL WTF", null, null);
+				update("LOL WTF -------------- " + script, null, null);
 			}
 		} catch(Exception | Error ex) {
 			this.fail(Types.stackTrace(ex));