Add model support for Onnx models in rank profiles

1 files changed, 3 insertions, 5 deletions

diff --git a/config-model/src/main/java/com/yahoo/searchdefinition/parser/ConvertParsedSchemas.java b/config-model/src/main/java/com/yahoo/searchdefinition/parser/ConvertParsedSchemas.java
index e56245d1332..4c32f11c20d 100644
--- a/config-model/src/main/java/com/yahoo/searchdefinition/parser/ConvertParsedSchemas.java
+++ b/config-model/src/main/java/com/yahoo/searchdefinition/parser/ConvertParsedSchemas.java
@@ -207,12 +207,10 @@ public class ConvertParsedSchemas {
         if (documentsOnly) {
             return; // skip ranking-only content, not used for document type generation
         }
-        for (var constant : parsed.getConstants()) {
+        for (var constant : parsed.getConstants())
             schema.add(constant);
-        }
-        for (var onnxModel : parsed.getOnnxModels()) {
-            schema.onnxModels().add(onnxModel);
-        }
+        for (var onnxModel : parsed.getOnnxModels())
+            schema.add(onnxModel);
         rankProfileRegistry.add(new DefaultRankProfile(schema, rankProfileRegistry));
         rankProfileRegistry.add(new UnrankedRankProfile(schema, rankProfileRegistry));
         var rankConverter = new ConvertParsedRanking(rankProfileRegistry);