diff options
author | Håvard Pettersen <havardpe@oath.com> | 2020-08-31 10:32:50 +0000 |
---|---|---|
committer | Håvard Pettersen <havardpe@oath.com> | 2020-09-01 12:45:13 +0000 |
commit | 3440f424ab11d8d8810b6f9785e6a8fad7271fe1 (patch) | |
tree | b769b0acf1485b4b18fd51efe556c8a6b583c7bf /searchcore/src/tests/proton/proton_configurer | |
parent | 3b8f7fdff4872bd010286753f6072ec492f14a48 (diff) |
handle onnx models config
Diffstat (limited to 'searchcore/src/tests/proton/proton_configurer')
-rw-r--r-- | searchcore/src/tests/proton/proton_configurer/proton_configurer_test.cpp | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/searchcore/src/tests/proton/proton_configurer/proton_configurer_test.cpp b/searchcore/src/tests/proton/proton_configurer/proton_configurer_test.cpp index 83706d966ae..6190177ac9d 100644 --- a/searchcore/src/tests/proton/proton_configurer/proton_configurer_test.cpp +++ b/searchcore/src/tests/proton/proton_configurer/proton_configurer_test.cpp @@ -19,6 +19,7 @@ #include <vespa/searchcore/proton/server/i_proton_disk_layout.h> #include <vespa/searchsummary/config/config-juniperrc.h> #include <vespa/searchcore/config/config-ranking-constants.h> +#include <vespa/searchcore/config/config-onnx-models.h> #include <vespa/vespalib/gtest/gtest.h> #include <vespa/searchcommon/common/schemaconfigurer.h> #include <vespa/vespalib/util/threadstackexecutor.h> @@ -44,12 +45,14 @@ using std::map; using search::index::Schema; using search::index::SchemaBuilder; using proton::matching::RankingConstants; +using proton::matching::OnnxModels; struct DBConfigFixture { using UP = std::unique_ptr<DBConfigFixture>; AttributesConfigBuilder _attributesBuilder; RankProfilesConfigBuilder _rankProfilesBuilder; RankingConstantsConfigBuilder _rankingConstantsBuilder; + OnnxModelsConfigBuilder _onnxModelsBuilder; IndexschemaConfigBuilder _indexschemaBuilder; SummaryConfigBuilder _summaryBuilder; SummarymapConfigBuilder _summarymapBuilder; @@ -70,6 +73,11 @@ struct DBConfigFixture { return std::make_shared<RankingConstants>(); } + OnnxModels::SP buildOnnxModels() + { + return std::make_shared<OnnxModels>(); + } + DocumentDBConfig::SP getConfig(int64_t generation, std::shared_ptr<DocumenttypesConfig> documentTypes, std::shared_ptr<const DocumentTypeRepo> repo, @@ -80,6 +88,7 @@ struct DBConfigFixture { (generation, std::make_shared<RankProfilesConfig>(_rankProfilesBuilder), buildRankingConstants(), + buildOnnxModels(), std::make_shared<IndexschemaConfig>(_indexschemaBuilder), std::make_shared<AttributesConfig>(_attributesBuilder), std::make_shared<SummaryConfig>(_summaryBuilder), |