diff options
author | Jo Kristian Bergum <bergum@yahooinc.com> | 2024-02-12 18:22:17 +0100 |
---|---|---|
committer | Jo Kristian Bergum <bergum@yahooinc.com> | 2024-02-12 18:22:17 +0100 |
commit | 1359ee25a46bd56bf65c7d2b9f789c42836e9b82 (patch) | |
tree | 614942a0ac6846b1050201161376640d40b4395f | |
parent | 7bd72abc858ce80328ad0da60f1472306f0feb49 (diff) |
preserve value type
-rw-r--r-- | indexinglanguage/src/main/java/com/yahoo/vespa/indexinglanguage/expressions/EmbedExpression.java | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/indexinglanguage/src/main/java/com/yahoo/vespa/indexinglanguage/expressions/EmbedExpression.java b/indexinglanguage/src/main/java/com/yahoo/vespa/indexinglanguage/expressions/EmbedExpression.java index 6e833ded9ba..1cfe8532c92 100644 --- a/indexinglanguage/src/main/java/com/yahoo/vespa/indexinglanguage/expressions/EmbedExpression.java +++ b/indexinglanguage/src/main/java/com/yahoo/vespa/indexinglanguage/expressions/EmbedExpression.java @@ -125,7 +125,7 @@ public class EmbedExpression extends Expression { String innerMappedDimension = targetType.mappedSubtype().dimensionNames().stream().filter(d -> !d.equals(outerMappedDimension)).findFirst().get(); String indexedDimension = targetType.indexedSubtype().dimensions().get(0).name(); long indexedDimensionSize = targetType.indexedSubtype().dimensions().get(0).size().get(); - var innerType = new TensorType.Builder().mapped(innerMappedDimension).indexed(indexedDimension,indexedDimensionSize).build(); + var innerType = new TensorType.Builder(targetType.valueType()).mapped(innerMappedDimension).indexed(indexedDimension,indexedDimensionSize).build(); int innerMappedDimensionIndex = innerType.indexOfDimensionAsInt(innerMappedDimension); int indexedDimensionIndex = innerType.indexOfDimensionAsInt(indexedDimension); for (int i = 0; i < input.size(); i++) { |