correct switch

bioimage-io · Oct 2, 2024 · a2809d1 · a2809d1
1 parent 88667e5
commit a2809d1
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/main/java/io/bioimage/modelrunner/pytorch/shm/ShmBuilder.java b/src/main/java/io/bioimage/modelrunner/pytorch/shm/ShmBuilder.java
@@ -61,18 +61,24 @@ private ShmBuilder()
      */
 	public static void build(NDArray  tensor, String memoryName) throws IllegalArgumentException, IOException
     {
+		System.out.println(tensor.getDataType().asNumpy());
 		switch (tensor.getDataType())
         {
             case UINT8:
             	buildFromTensorUByte(tensor, memoryName);
+            	break;
             case INT32:
             	buildFromTensorInt(tensor, memoryName);
+            	break;
             case FLOAT32:
             	buildFromTensorFloat(tensor, memoryName);
+            	break;
             case FLOAT64:
             	buildFromTensorDouble(tensor, memoryName);
+            	break;
             case INT64:
             	buildFromTensorLong(tensor, memoryName);
+            	break;
             default:
                 throw new IllegalArgumentException("Unsupported tensor type: " + tensor.getDataType().asNumpy());
         }