diff options
Diffstat (limited to 'searchlib')
-rwxr-xr-x | searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/rule/ReferenceNode.java | 14 |
1 files changed, 11 insertions, 3 deletions
diff --git a/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/rule/ReferenceNode.java b/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/rule/ReferenceNode.java index cd4eb21d1f2..b9b377dc0ec 100755 --- a/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/rule/ReferenceNode.java +++ b/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/rule/ReferenceNode.java @@ -60,6 +60,10 @@ public final class ReferenceNode extends CompositeNode { @Override public String toString(SerializationContext context, Deque<String> path, CompositeNode parent) { + return toString(context, path, true); + } + + private String toString(SerializationContext context, Deque<String> path, boolean includeOutput) { if (path == null) path = new ArrayDeque<>(); String myName = this.name; @@ -101,14 +105,18 @@ public final class ReferenceNode extends CompositeNode { } ret.append(")"); } - ret.append(myOutput != null ? "." + myOutput : ""); + if (includeOutput) + ret.append(myOutput != null ? "." + myOutput : ""); return ret.toString(); } @Override public TensorType type(TypeContext context) { - // Don't support outputs of different type, for simplicity - TensorType type = context.getType(toString()); + // Ensure base name (excluding output exists, + // but don't support outputs of different tensor types (not used, so no need) + String name = toString(new SerializationContext(), null, false); + TensorType type = context.getType(name); + if (type == null) throw new IllegalArgumentException("Unknown feature '" + toString() + "'"); return type; |