diff options
author | Henning Baldersheim <balder@yahoo-inc.com> | 2019-10-31 15:03:49 +0100 |
---|---|---|
committer | Henning Baldersheim <balder@yahoo-inc.com> | 2019-10-31 15:11:20 +0100 |
commit | 479b3fdd1e2d9e14e3f2ad666cac91ca8de91816 (patch) | |
tree | e8e2dba799dc59fc6acf1767c3f5235171767701 /config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/RankProfileTransformContext.java | |
parent | 6ab942251da3c3b4c36cfd1fa1adb12984534137 (diff) |
Precompute the fixed set of feature types once.
Diffstat (limited to 'config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/RankProfileTransformContext.java')
-rw-r--r-- | config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/RankProfileTransformContext.java | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/RankProfileTransformContext.java b/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/RankProfileTransformContext.java index c76b8536ea0..a12b06624cf 100644 --- a/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/RankProfileTransformContext.java +++ b/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/RankProfileTransformContext.java @@ -4,8 +4,10 @@ package com.yahoo.searchdefinition.expressiontransforms; import ai.vespa.rankingexpression.importer.configmodelview.ImportedMlModels; import com.yahoo.search.query.profile.QueryProfileRegistry; import com.yahoo.searchdefinition.RankProfile; +import com.yahoo.searchlib.rankingexpression.Reference; import com.yahoo.searchlib.rankingexpression.evaluation.Value; import com.yahoo.searchlib.rankingexpression.transform.TransformContext; +import com.yahoo.tensor.TensorType; import java.util.HashMap; import java.util.Map; @@ -25,10 +27,11 @@ public class RankProfileTransformContext extends TransformContext { public RankProfileTransformContext(RankProfile rankProfile, QueryProfileRegistry queryProfiles, + Map<Reference, TensorType> featureTypes, ImportedMlModels importedModels, Map<String, Value> constants, Map<String, RankProfile.RankingExpressionFunction> inlineFunctions) { - super(constants, rankProfile.typeContext(queryProfiles)); + super(constants, rankProfile.typeContext(queryProfiles, featureTypes)); this.rankProfile = rankProfile; this.queryProfiles = queryProfiles; this.importedModels = importedModels; |