diff options
Diffstat (limited to 'config-model/src/main/java/com/yahoo/schema/processing/OnnxModelConfigGenerator.java')
-rw-r--r-- | config-model/src/main/java/com/yahoo/schema/processing/OnnxModelConfigGenerator.java | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/config-model/src/main/java/com/yahoo/schema/processing/OnnxModelConfigGenerator.java b/config-model/src/main/java/com/yahoo/schema/processing/OnnxModelConfigGenerator.java index ce56a4320d3..338977fa679 100644 --- a/config-model/src/main/java/com/yahoo/schema/processing/OnnxModelConfigGenerator.java +++ b/config-model/src/main/java/com/yahoo/schema/processing/OnnxModelConfigGenerator.java @@ -50,6 +50,9 @@ public class OnnxModelConfigGenerator extends Processor { if (profile.getSecondPhaseRanking() != null) { process(profile.getSecondPhaseRanking().getRoot(), profile); } + if (profile.getGlobalPhaseRanking() != null) { + process(profile.getGlobalPhaseRanking().getRoot(), profile); + } for (Map.Entry<String, RankProfile.RankingExpressionFunction> function : profile.getFunctions().entrySet()) { process(function.getValue().function().getBody().getRoot(), profile); } |