From 19f9c783c2f1ca136a6ed874656e0a2c93b4adca Mon Sep 17 00:00:00 2001 From: Lester Solbakken Date: Mon, 10 Jan 2022 14:34:04 +0100 Subject: onnxModel to onnx in summary/matchfeatures --- config-model/src/main/java/com/yahoo/schema/RankProfile.java | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'config-model/src/main/java/com/yahoo/schema/RankProfile.java') diff --git a/config-model/src/main/java/com/yahoo/schema/RankProfile.java b/config-model/src/main/java/com/yahoo/schema/RankProfile.java index 5479ecf323f..56786c733ec 100644 --- a/config-model/src/main/java/com/yahoo/schema/RankProfile.java +++ b/config-model/src/main/java/com/yahoo/schema/RankProfile.java @@ -1089,11 +1089,11 @@ public class RankProfile implements Cloneable { Map inputTypes = resolveOnnxInputTypes(model, context); TensorType defaultOutputType = model.getTensorType(model.getDefaultOutput(), inputTypes); - context.setType(new Reference("onnxModel", args, null), defaultOutputType); + context.setType(new Reference("onnx", args, null), defaultOutputType); for (Map.Entry mapping : model.getOutputMap().entrySet()) { TensorType type = model.getTensorType(mapping.getKey(), inputTypes); - context.setType(new Reference("onnxModel", args, mapping.getValue()), type); + context.setType(new Reference("onnx", args, mapping.getValue()), type); } } return context; -- cgit v1.2.3