Use correct value type for onnx range operator

1 files changed, 1 insertions, 1 deletions

diff --git a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Range.java b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Range.java
index 6df686cf910..18018a1a73a 100644
--- a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Range.java
+++ b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Range.java
@@ -51,7 +51,7 @@ public class Range extends IntermediateOperation {
         delta = getConstantInput(2, "delta");
         elements = (long) Math.ceil((limit - start) / delta);
 
-        OrderedTensorType type = new OrderedTensorType.Builder()
+        OrderedTensorType type = new OrderedTensorType.Builder(inputs.get(0).type().get().type().valueType())
                 .add(TensorType.Dimension.indexed(vespaName(), elements))
                 .build();
         return type;