diff options
author | Jon Bratseth <bratseth@oath.com> | 2018-09-24 16:42:29 -0700 |
---|---|---|
committer | Jon Bratseth <bratseth@oath.com> | 2018-09-24 16:42:29 -0700 |
commit | 6162ec099bc06ac7ba7a82f2ca59aa1e869a6c96 (patch) | |
tree | 27e337ceef6a3f6a9c1dae51a670615713f03661 /config-model | |
parent | edfd60817bd96eec9175f0f9a35ad8a3196374e6 (diff) |
Serialize argument type information when available
Diffstat (limited to 'config-model')
-rw-r--r-- | config-model/src/main/java/com/yahoo/searchdefinition/derived/RawRankProfile.java | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/config-model/src/main/java/com/yahoo/searchdefinition/derived/RawRankProfile.java b/config-model/src/main/java/com/yahoo/searchdefinition/derived/RawRankProfile.java index f575d1178c0..279b5334187 100644 --- a/config-model/src/main/java/com/yahoo/searchdefinition/derived/RawRankProfile.java +++ b/config-model/src/main/java/com/yahoo/searchdefinition/derived/RawRankProfile.java @@ -13,6 +13,7 @@ import com.yahoo.searchlib.rankingexpression.integration.ml.ImportedModels; import com.yahoo.searchlib.rankingexpression.parser.ParseException; import com.yahoo.searchlib.rankingexpression.rule.ReferenceNode; import com.yahoo.searchlib.rankingexpression.rule.SerializationContext; +import com.yahoo.tensor.TensorType; import com.yahoo.vespa.config.search.RankProfilesConfig; import java.nio.charset.Charset; @@ -208,8 +209,10 @@ public class RawRankProfile implements RankProfilesConfig.Producer { String expressionString = e.getValue().function().getBody().getRoot().toString(new StringBuilder(), context, null, null).toString(); context.addFunctionSerialization(RankingExpression.propertyName(e.getKey()), expressionString); + for (Map.Entry<String, TensorType> argumentType : e.getValue().function().argumentTypes().entrySet()) + context.addArgumentTypeSerialization(e.getKey(), argumentType.getKey(), argumentType.getValue()); if (e.getValue().function().returnType().isPresent()) - context.addFunctionTypeSerialization(RankingExpression.propertyTypeName(e.getKey()), e.getValue().function().returnType().get().toString()); + context.addFunctionTypeSerialization(e.getKey(), e.getValue().function().returnType().get()); else if (e.getValue().function().arguments().isEmpty()) throw new IllegalStateException("Type of function '" + e.getKey() + "' is not resolved"); } |