improve error message

bioimage-io · Mar 25, 2024 · 3ff4c45 · 3ff4c45
1 parent ebf64f9
commit 3ff4c45
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/main/java/io/bioimage/modelrunner/onnx/tensor/TensorBuilder.java b/src/main/java/io/bioimage/modelrunner/onnx/tensor/TensorBuilder.java
@@ -126,7 +126,7 @@ private static OnnxTensor buildByte(RandomAccessibleInterval<ByteType> tensor, O
 		long[] ogShape = tensor.dimensionsAsLongArray();
 		if (CommonUtils.int32Overflows(ogShape, 1))
 			throw new IllegalArgumentException("Provided tensor with shape " + Arrays.toString(ogShape) 
-								+ " is too big. Max number of elements per tensor supported: " + Integer.MAX_VALUE);
+								+ " is too big. Max number of elements per byte tensor supported: " + Integer.MAX_VALUE);
 		tensor = Utils.transpose(tensor);
 		long[] tensorShape = tensor.dimensionsAsLongArray();
 		int size = 1;
@@ -164,7 +164,7 @@ private static OnnxTensor buildInt(RandomAccessibleInterval<IntType> tensor, Ort
 		long[] ogShape = tensor.dimensionsAsLongArray();
 		if (CommonUtils.int32Overflows(ogShape, 4))
 			throw new IllegalArgumentException("Provided tensor with shape " + Arrays.toString(ogShape) 
-								+ " is too big. Max number of elements per tensor supported: " + Integer.MAX_VALUE / 4);
+								+ " is too big. Max number of elements per int tensor supported: " + Integer.MAX_VALUE / 4);
 		tensor = Utils.transpose(tensor);
 		long[] tensorShape = tensor.dimensionsAsLongArray();
 		int size = 1;
@@ -202,7 +202,7 @@ private static OnnxTensor buildFloat(RandomAccessibleInterval<FloatType> tensor,
 		long[] ogShape = tensor.dimensionsAsLongArray();
 		if (CommonUtils.int32Overflows(ogShape, 4))
 			throw new IllegalArgumentException("Provided tensor with shape " + Arrays.toString(ogShape) 
-								+ " is too big. Max number of elements per tensor supported: " + Integer.MAX_VALUE / 4);
+								+ " is too big. Max number of elements per float tensor supported: " + Integer.MAX_VALUE / 4);
 		tensor = Utils.transpose(tensor);
 		long[] tensorShape = tensor.dimensionsAsLongArray();
 		int size = 1;
@@ -240,7 +240,7 @@ private static OnnxTensor buildDouble(RandomAccessibleInterval<DoubleType> tenso
 		long[] ogShape = tensor.dimensionsAsLongArray();
 		if (CommonUtils.int32Overflows(ogShape, 8))
 			throw new IllegalArgumentException("Provided tensor with shape " + Arrays.toString(ogShape) 
-								+ " is too big. Max number of elements per tensor supported: " + Integer.MAX_VALUE / 8);
+								+ " is too big. Max number of elements per double tensor supported: " + Integer.MAX_VALUE / 8);
 		tensor = Utils.transpose(tensor);
 		long[] tensorShape = tensor.dimensionsAsLongArray();
 		int size = 1;