diff options
author | Henning Baldersheim <balder@yahoo-inc.com> | 2021-05-23 17:55:57 +0200 |
---|---|---|
committer | Henning Baldersheim <balder@yahoo-inc.com> | 2021-05-23 17:55:57 +0200 |
commit | 3579fbb70363aaf419bdb1465dfe8a04047b4350 (patch) | |
tree | f0ef3957db72c061c4d49c3e4bd9c5d1c5673805 /config-model | |
parent | 32777b8b9544e775689ba25c63280fa91a00dadd (diff) |
provide empty config
Diffstat (limited to 'config-model')
-rw-r--r-- | config-model/src/test/java/com/yahoo/vespa/model/ml/ModelEvaluationTest.java | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/config-model/src/test/java/com/yahoo/vespa/model/ml/ModelEvaluationTest.java b/config-model/src/test/java/com/yahoo/vespa/model/ml/ModelEvaluationTest.java index d0196ace766..e32ad274325 100644 --- a/config-model/src/test/java/com/yahoo/vespa/model/ml/ModelEvaluationTest.java +++ b/config-model/src/test/java/com/yahoo/vespa/model/ml/ModelEvaluationTest.java @@ -17,6 +17,7 @@ import com.yahoo.tensor.TensorType; import com.yahoo.vespa.config.search.RankProfilesConfig; import com.yahoo.vespa.config.search.core.OnnxModelsConfig; import com.yahoo.vespa.config.search.core.RankingConstantsConfig; +import com.yahoo.vespa.config.search.core.RankingExpressionsConfig; import com.yahoo.vespa.model.VespaModel; import com.yahoo.vespa.model.container.ApplicationContainerCluster; import org.junit.Test; @@ -96,6 +97,9 @@ public class ModelEvaluationTest { cluster.getConfig(cb); RankingConstantsConfig constantsConfig = new RankingConstantsConfig(cb); + RankingExpressionsConfig.Builder ce = new RankingExpressionsConfig.Builder(); + RankingExpressionsConfig expressionsConfig = ce.build(); + OnnxModelsConfig.Builder ob = new OnnxModelsConfig.Builder(); cluster.getConfig(ob); OnnxModelsConfig onnxModelsConfig = new OnnxModelsConfig(ob); @@ -114,7 +118,7 @@ public class ModelEvaluationTest { assertEquals(profile, sb.toString()); ModelsEvaluator evaluator = new ModelsEvaluator(new ToleratingMissingConstantFilesRankProfilesConfigImporter(MockFileAcquirer.returnFile(null)) - .importFrom(config, constantsConfig, onnxModelsConfig)); + .importFrom(config, constantsConfig, expressionsConfig, onnxModelsConfig)); assertEquals(4, evaluator.models().size()); |