diff options
author | HÃ¥vard Pettersen <3535158+havardpe@users.noreply.github.com> | 2020-09-02 10:23:48 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-09-02 10:23:48 +0200 |
commit | 014dcfd819b89c8321e9d357e2418067b0311907 (patch) | |
tree | 84570305c1f6fa32364eacf0bf6d73eb147e83e5 /searchlib/src/tests/features | |
parent | c9f22e35d3e13d0d07e091f49548291d21d98d13 (diff) | |
parent | 3440f424ab11d8d8810b6f9785e6a8fad7271fe1 (diff) |
Merge pull request #14227 from vespa-engine/havardpe/handle-onnx-models-config
handle onnx models config
Diffstat (limited to 'searchlib/src/tests/features')
-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); |