From 8d8f523c32fab2bd396f960496459eb079b03872 Mon Sep 17 00:00:00 2001
From: Arne Juul <arnej@yahooinc.com>
Date: Thu, 16 Feb 2023 13:24:04 +0000
Subject: deeper processing of onnx inputs

---
 .../com/yahoo/schema/expressiontransforms/InputRecorder.java  | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

(limited to 'config-model/src/main')

diff --git a/config-model/src/main/java/com/yahoo/schema/expressiontransforms/InputRecorder.java b/config-model/src/main/java/com/yahoo/schema/expressiontransforms/InputRecorder.java
index 4e7988a2006..b0f63ebb732 100644
--- a/config-model/src/main/java/com/yahoo/schema/expressiontransforms/InputRecorder.java
+++ b/config-model/src/main/java/com/yahoo/schema/expressiontransforms/InputRecorder.java
@@ -3,13 +3,16 @@ package com.yahoo.schema.expressiontransforms;
 
 import com.yahoo.schema.FeatureNames;
 import com.yahoo.schema.RankProfile;
+import com.yahoo.searchlib.rankingexpression.RankingExpression;
 import com.yahoo.searchlib.rankingexpression.Reference;
+import com.yahoo.searchlib.rankingexpression.parser.ParseException;
 import com.yahoo.searchlib.rankingexpression.rule.CompositeNode;
 import com.yahoo.searchlib.rankingexpression.rule.ConstantNode;
 import com.yahoo.searchlib.rankingexpression.rule.ExpressionNode;
 import com.yahoo.searchlib.rankingexpression.rule.ReferenceNode;
 import com.yahoo.searchlib.rankingexpression.transform.ExpressionTransformer;
 
+import java.io.StringReader;
 import java.util.Set;
 
 /**
@@ -83,7 +86,13 @@ public class InputRecorder extends ExpressionTransformer<RankProfileTransformCon
                 throw new IllegalArgumentException("missing onnx model: " + arg);
             }
             for (String onnxInput : model.getInputMap().values()) {
-                neededInputs.add(onnxInput);
+                var reader = new StringReader(onnxInput);
+                try {
+                    var asExpression = new RankingExpression(reader);
+                    transform(asExpression.getRoot(), context);
+                } catch (ParseException e) {
+                    throw new IllegalArgumentException("illegal onnx input '" + onnxInput + "': " + e.getMessage());
+                }
             }
             return;
         }
-- 
cgit v1.2.3