diff options
author | Bjørn Christian Seime <bjorncs@yahooinc.com> | 2023-04-13 11:34:41 +0200 |
---|---|---|
committer | Bjørn Christian Seime <bjorncs@yahooinc.com> | 2023-04-13 11:34:41 +0200 |
commit | a68180b364aa4bbb08a8cc0220a6e20fe7b4404d (patch) | |
tree | 52397579824e17163c22a498221a537fe35c5465 /config-model/src/main/java/com/yahoo/schema/expressiontransforms/InputRecorder.java | |
parent | 7e1b7baba3f2f723405985d636089650a521f5d7 (diff) |
Ignore input also listed in intializers when parsing metadata from model
Diffstat (limited to 'config-model/src/main/java/com/yahoo/schema/expressiontransforms/InputRecorder.java')
-rw-r--r-- | config-model/src/main/java/com/yahoo/schema/expressiontransforms/InputRecorder.java | 6 |
1 files changed, 1 insertions, 5 deletions
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 7f578f07fe3..5d3624cd3d3 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 @@ -120,11 +120,7 @@ public class InputRecorder extends ExpressionTransformer<InputRecorderContext> { if (model == null) { throw new IllegalArgumentException("missing onnx model: " + arg); } - model.getInputMap().forEach((onnxName, onnxInput) -> { - if (model.getInitializers().contains(onnxName)) { - log.fine(() -> "For input '%s': skipping name '%s' as it's an initializer".formatted(onnxInput, onnxName)); - return; - } + model.getInputMap().forEach((__, onnxInput) -> { var reader = new StringReader(onnxInput); try { var asExpression = new RankingExpression(reader); |