summaryrefslogtreecommitdiffstats
path: root/container-search
diff options
context:
space:
mode:
authorBjørn Christian Seime <bjorncs@vespa.ai>2024-05-06 13:50:10 +0200
committerBjørn Christian Seime <bjorncs@vespa.ai>2024-05-06 13:50:10 +0200
commit66c699e65e875233f6fd8898fceb7eebd4c4546a (patch)
tree4f616654c4531d2a2ae928f2b0342871511bc193 /container-search
parentbf6390b8dff4d1a5569e531ec68da8cd33b7ad73 (diff)
Populate significance properties from config
Diffstat (limited to 'container-search')
-rw-r--r--container-search/src/main/java/com/yahoo/search/schema/SchemaInfoConfigurer.java7
-rw-r--r--container-search/src/test/java/com/yahoo/search/schema/SchemaInfoTester.java4
2 files changed, 7 insertions, 4 deletions
diff --git a/container-search/src/main/java/com/yahoo/search/schema/SchemaInfoConfigurer.java b/container-search/src/main/java/com/yahoo/search/schema/SchemaInfoConfigurer.java
index d28c2db2b9e..77f27d3d411 100644
--- a/container-search/src/main/java/com/yahoo/search/schema/SchemaInfoConfigurer.java
+++ b/container-search/src/main/java/com/yahoo/search/schema/SchemaInfoConfigurer.java
@@ -22,9 +22,10 @@ class SchemaInfoConfigurer {
Schema.Builder builder = new Schema.Builder(schemaInfoConfig.name());
for (var profileConfig : schemaInfoConfig.rankprofile()) {
- RankProfile.Builder profileBuilder = new RankProfile.Builder(profileConfig.name());
- profileBuilder.setHasSummaryFeatures(profileConfig.hasSummaryFeatures());
- profileBuilder.setHasRankFeatures(profileConfig.hasRankFeatures());
+ RankProfile.Builder profileBuilder = new RankProfile.Builder(profileConfig.name())
+ .setHasSummaryFeatures(profileConfig.hasSummaryFeatures())
+ .setHasRankFeatures(profileConfig.hasRankFeatures())
+ .setUseSignificanceModel(profileConfig.significance().useModel());
for (var inputConfig : profileConfig.input())
profileBuilder.addInput(inputConfig.name(), RankProfile.InputType.fromSpec(inputConfig.type()));
builder.add(profileBuilder.build());
diff --git a/container-search/src/test/java/com/yahoo/search/schema/SchemaInfoTester.java b/container-search/src/test/java/com/yahoo/search/schema/SchemaInfoTester.java
index 3e98b911fc8..2ba399cf42d 100644
--- a/container-search/src/test/java/com/yahoo/search/schema/SchemaInfoTester.java
+++ b/container-search/src/test/java/com/yahoo/search/schema/SchemaInfoTester.java
@@ -77,6 +77,7 @@ public class SchemaInfoTester {
.addInput("query(myTensor1)", InputType.fromSpec("tensor(x[10])"))
.build())
.add(new RankProfile.Builder("bOnly")
+ .setUseSignificanceModel(true)
.addInput("query(myTensor1)", InputType.fromSpec("tensor(a{},b{})"))
.build())
.build());
@@ -129,7 +130,8 @@ public class SchemaInfoTester {
rankProfileInconsistentB.input(new SchemaInfoConfig.Schema.Rankprofile.Input.Builder().name("query(myTensor1)").type("tensor(x[10])"));
schemaB.rankprofile(rankProfileInconsistentB);
var rankProfileBOnly = new SchemaInfoConfig.Schema.Rankprofile.Builder();
- rankProfileBOnly.name("bOnly");
+ rankProfileBOnly.name("bOnly")
+ .significance(new SchemaInfoConfig.Schema.Rankprofile.Significance.Builder().useModel(true));
rankProfileBOnly.input(new SchemaInfoConfig.Schema.Rankprofile.Input.Builder().name("query(myTensor1)").type("tensor(a{},b{})"));
schemaB.rankprofile(rankProfileBOnly);