diff options
author | Lester Solbakken <lesters@users.noreply.github.com> | 2020-10-27 21:42:26 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-10-27 21:42:26 +0100 |
commit | 47d59cef3b5f06821b8d475f23b36d0f9cdcc0c6 (patch) | |
tree | 733506db79728b238bb9687a9e5bab20f78204a3 /config-model/src/main/java/com/yahoo/searchdefinition/derived/RankProfileList.java | |
parent | 4d5f1ce293c680be6a0050a0a1606a902003465d (diff) | |
parent | de5e15528fca545b2a9ccbb1386c5590e11fa383 (diff) |
Merge pull request #15049 from vespa-engine/revert-15028-lesters/revert-revert-resolve-onnx-model-types
Revert "Lesters/revert revert resolve onnx model types"
Diffstat (limited to 'config-model/src/main/java/com/yahoo/searchdefinition/derived/RankProfileList.java')
-rw-r--r-- | config-model/src/main/java/com/yahoo/searchdefinition/derived/RankProfileList.java | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/config-model/src/main/java/com/yahoo/searchdefinition/derived/RankProfileList.java b/config-model/src/main/java/com/yahoo/searchdefinition/derived/RankProfileList.java index 22a32c8fd65..84442fedc48 100644 --- a/config-model/src/main/java/com/yahoo/searchdefinition/derived/RankProfileList.java +++ b/config-model/src/main/java/com/yahoo/searchdefinition/derived/RankProfileList.java @@ -126,8 +126,8 @@ public class RankProfileList extends Derived implements RankProfilesConfig.Produ OnnxModelsConfig.Model.Builder modelBuilder = new OnnxModelsConfig.Model.Builder(); modelBuilder.name(model.getName()); modelBuilder.fileref(model.getFileReference()); - model.getInputMap().forEach((name, source) -> modelBuilder.input(new OnnxModelsConfig.Model.Input.Builder().name(name).source(source))); - model.getOutputMap().forEach((name, as) -> modelBuilder.output(new OnnxModelsConfig.Model.Output.Builder().name(name).as(as))); + model.getInputMap().forEach(mapper -> modelBuilder.input(new OnnxModelsConfig.Model.Input.Builder().name(mapper.getOnnxName()).source(mapper.getVespaName()))); + model.getOutputMap().forEach(mapper -> modelBuilder.output(new OnnxModelsConfig.Model.Output.Builder().name(mapper.getOnnxName()).as(mapper.getVespaName()))); builder.model(modelBuilder); } } |