specialize TensorFunction etc on Reference

1 files changed, 2 insertions, 2 deletions

diff --git a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Split.java b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Split.java
index 2e586b38c71..6f720716adb 100644
--- a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Split.java
+++ b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Split.java
@@ -84,7 +84,7 @@ public class Split extends IntermediateOperation {
     }
 
     @Override
-    protected TensorFunction lazyGetFunction() {
+    protected TensorFunction<Reference> lazyGetFunction() {
         if (!allInputFunctionsPresent(1)) return null;
 
         IntermediateOperation input = inputs.get(0);
@@ -104,7 +104,7 @@ public class Split extends IntermediateOperation {
         com.yahoo.tensor.functions.Slice<Reference> sliceIndices = new com.yahoo.tensor.functions.Slice<>(inputIndices, dimensionValues);
         ExpressionNode sliceExpression = new TensorFunctionNode(sliceIndices);
 
-        TensorFunction generate = Generate.bound(type.type(), wrapScalar(sliceExpression));
+        TensorFunction<Reference> generate = Generate.bound(type.type(), wrapScalar(sliceExpression));
         return generate;
     }