summaryrefslogtreecommitdiffstats
path: root/searchlib
diff options
context:
space:
mode:
Diffstat (limited to 'searchlib')
-rw-r--r--searchlib/abi-spec.json1
-rwxr-xr-xsearchlib/src/main/java/com/yahoo/searchlib/rankingexpression/ExpressionFunction.java1
2 files changed, 2 insertions, 0 deletions
diff --git a/searchlib/abi-spec.json b/searchlib/abi-spec.json
index 16c1b2c0e7d..5413907e967 100644
--- a/searchlib/abi-spec.json
+++ b/searchlib/abi-spec.json
@@ -277,6 +277,7 @@
"public java.util.List arguments()",
"public com.yahoo.searchlib.rankingexpression.RankingExpression getBody()",
"public java.util.Map argumentTypes()",
+ "public com.yahoo.tensor.TensorType getArgumentType(java.lang.String)",
"public java.util.Optional returnType()",
"public com.yahoo.searchlib.rankingexpression.ExpressionFunction withName(java.lang.String)",
"public com.yahoo.searchlib.rankingexpression.ExpressionFunction withBody(com.yahoo.searchlib.rankingexpression.RankingExpression)",
diff --git a/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/ExpressionFunction.java b/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/ExpressionFunction.java
index 241a53fb458..171151bfdf4 100755
--- a/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/ExpressionFunction.java
+++ b/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/ExpressionFunction.java
@@ -83,6 +83,7 @@ public class ExpressionFunction {
/** Returns the types of the arguments of this, if specified. The keys of this may be any subset of the arguments */
public Map<String, TensorType> argumentTypes() { return argumentTypes; }
+ public TensorType getArgumentType(String argumentName) { return argumentTypes.get(argumentName); }
/** Returns the return type of this, or empty if not specified */
public Optional<TensorType> returnType() { return returnType; }