Make tensor addresses integer based instead of as strings.

Positive numbers are direct indexes, while strings that does not represent numbers are enumerated
and represented with negative integers.

1 files changed, 1 insertions, 2 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 55863f51ea7..b3f2f265900 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,7 +17,6 @@ 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;
@@ -141,7 +140,7 @@ public class TensorFunctionNode extends CompositeNode {
                     labels[allDimensionsIndex++] = mappedDimensionLabel;
             }
 
-            receivingMap.put(StringTensorAddress.unsafeOf(labels), wrapScalar(node));
+            receivingMap.put(TensorAddress.of(labels), wrapScalar(node));
         }
     }