diff options
Diffstat (limited to 'config-model/src/main/java/com/yahoo/searchdefinition')
3 files changed, 7 insertions, 7 deletions
diff --git a/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/OnnxModelTransformer.java b/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/OnnxModelTransformer.java index faf3c8085d8..6d9f4cdec92 100644 --- a/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/OnnxModelTransformer.java +++ b/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/OnnxModelTransformer.java @@ -135,7 +135,7 @@ public class OnnxModelTransformer extends ExpressionTransformer<RankProfileTrans public static String asString(ExpressionNode node) { if ( ! (node instanceof ConstantNode)) throw new IllegalArgumentException("Expected a constant string as argument, but got '" + node); - return stripQuotes(((ConstantNode)node).sourceString()); + return stripQuotes(node.toString()); } } diff --git a/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/TokenTransformer.java b/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/TokenTransformer.java index cd56de18527..dbb32d88ef6 100644 --- a/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/TokenTransformer.java +++ b/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/TokenTransformer.java @@ -43,11 +43,11 @@ import static com.yahoo.searchlib.rankingexpression.rule.TensorFunctionNode.wrap */ public class TokenTransformer extends ExpressionTransformer<RankProfileTransformContext> { - static private final ConstantNode ZERO = new ConstantNode(new DoubleValue(0.0), "0"); - static private final ConstantNode ONE = new ConstantNode(new DoubleValue(1.0), "1"); - static private final ConstantNode TWO = new ConstantNode(new DoubleValue(2.0), "2"); - static private final ConstantNode CLS = new ConstantNode(new DoubleValue(101), "101"); - static private final ConstantNode SEP = new ConstantNode(new DoubleValue(102), "102"); + static private final ConstantNode ZERO = new ConstantNode(new DoubleValue(0.0)); + static private final ConstantNode ONE = new ConstantNode(new DoubleValue(1.0)); + static private final ConstantNode TWO = new ConstantNode(new DoubleValue(2.0)); + static private final ConstantNode CLS = new ConstantNode(new DoubleValue(101)); + static private final ConstantNode SEP = new ConstantNode(new DoubleValue(102)); @Override public ExpressionNode transform(ExpressionNode node, RankProfileTransformContext context) { diff --git a/config-model/src/main/java/com/yahoo/searchdefinition/processing/OnnxModelConfigGenerator.java b/config-model/src/main/java/com/yahoo/searchdefinition/processing/OnnxModelConfigGenerator.java index c7c1db3b862..fdbde08d926 100644 --- a/config-model/src/main/java/com/yahoo/searchdefinition/processing/OnnxModelConfigGenerator.java +++ b/config-model/src/main/java/com/yahoo/searchdefinition/processing/OnnxModelConfigGenerator.java @@ -74,7 +74,7 @@ public class OnnxModelConfigGenerator extends Processor { if (feature.getArguments().size() > 0) { if (feature.getArguments().expressions().get(0) instanceof ConstantNode) { ConstantNode node = (ConstantNode) feature.getArguments().expressions().get(0); - String path = OnnxModelTransformer.stripQuotes(node.sourceString()); + String path = OnnxModelTransformer.stripQuotes(node.toString()); String modelConfigName = OnnxModelTransformer.asValidIdentifier(path); // Only add the configuration if the model can actually be found. |