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 /searchlib/src/tests/features/onnx_feature/onnx_feature_test.cpp | |
parent | 3b8f7fdff4872bd010286753f6072ec492f14a48 (diff) |
handle onnx models config
Diffstat (limited to 'searchlib/src/tests/features/onnx_feature/onnx_feature_test.cpp')
-rw-r--r-- | searchlib/src/tests/features/onnx_feature/onnx_feature_test.cpp | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/searchlib/src/tests/features/onnx_feature/onnx_feature_test.cpp b/searchlib/src/tests/features/onnx_feature/onnx_feature_test.cpp index 7a200a46ab2..826984832f6 100644 --- a/searchlib/src/tests/features/onnx_feature/onnx_feature_test.cpp +++ b/searchlib/src/tests/features/onnx_feature/onnx_feature_test.cpp @@ -58,9 +58,7 @@ struct OnnxFeatureTest : ::testing::Test { indexEnv.getProperties().add(expr_name, expr); } void add_onnx(const vespalib::string &name, const vespalib::string &file) { - vespalib::string feature_name = onnx_feature(name); - vespalib::string file_name = feature_name + ".fileref"; - indexEnv.getProperties().add(file_name, file); + indexEnv.addOnnxModel(name, file); } void compile(const vespalib::string &seed) { resolver->addSeed(seed); |