Merge pull request #29965 from vespa-engine/balder/make-tensor-impl-package

Balder/make tensor impl package

1 files changed, 2 insertions, 1 deletions

diff --git a/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/rule/TensorFunctionNode.java b/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/rule/TensorFunctionNode.java
index b3f2f265900..55863f51ea7 100644
--- a/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/rule/TensorFunctionNode.java
+++ b/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/rule/TensorFunctionNode.java
@@ -17,6 +17,7 @@ import com.yahoo.tensor.functions.PrimitiveTensorFunction;
 import com.yahoo.tensor.functions.ScalarFunction;
 import com.yahoo.tensor.functions.TensorFunction;
 import com.yahoo.tensor.functions.ToStringContext;
+import com.yahoo.tensor.impl.StringTensorAddress;
 
 import java.util.ArrayList;
 import java.util.Collections;
@@ -140,7 +141,7 @@ public class TensorFunctionNode extends CompositeNode {
                     labels[allDimensionsIndex++] = mappedDimensionLabel;
             }
 
-            receivingMap.put(TensorAddress.of(labels), wrapScalar(node));
+            receivingMap.put(StringTensorAddress.unsafeOf(labels), wrapScalar(node));
         }
     }