diff options
Diffstat (limited to 'model-evaluation/src/test/java/ai/vespa/models/handler/OnnxEvaluationHandlerTest.java')
-rw-r--r-- | model-evaluation/src/test/java/ai/vespa/models/handler/OnnxEvaluationHandlerTest.java | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/model-evaluation/src/test/java/ai/vespa/models/handler/OnnxEvaluationHandlerTest.java b/model-evaluation/src/test/java/ai/vespa/models/handler/OnnxEvaluationHandlerTest.java index 6cfda4d8ce8..97692de56ef 100644 --- a/model-evaluation/src/test/java/ai/vespa/models/handler/OnnxEvaluationHandlerTest.java +++ b/model-evaluation/src/test/java/ai/vespa/models/handler/OnnxEvaluationHandlerTest.java @@ -11,6 +11,7 @@ import com.yahoo.tensor.Tensor; 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 org.junit.BeforeClass; import org.junit.Test; @@ -122,6 +123,8 @@ public class OnnxEvaluationHandlerTest { RankProfilesConfig.class).getConfig(""); RankingConstantsConfig constantsConfig = new ConfigGetter<>(new FileSource(configDir.append("ranking-constants.cfg").toFile()), RankingConstantsConfig.class).getConfig(""); + RankingExpressionsConfig expressionsConfig = new ConfigGetter<>(new FileSource(configDir.append("ranking-expressions.cfg").toFile()), + RankingExpressionsConfig.class).getConfig(""); OnnxModelsConfig onnxModelsConfig = new ConfigGetter<>(new FileSource(configDir.append("onnx-models.cfg").toFile()), OnnxModelsConfig.class).getConfig(""); @@ -131,7 +134,7 @@ public class OnnxEvaluationHandlerTest { } FileAcquirer fileAcquirer = MockFileAcquirer.returnFiles(fileMap); - return new ModelsEvaluator(config, constantsConfig, onnxModelsConfig, fileAcquirer); + return new ModelsEvaluator(config, constantsConfig, expressionsConfig, onnxModelsConfig, fileAcquirer); } } |