Import Tensorflow models vis ONNX conversion

1 files changed, 3 insertions, 3 deletions

diff --git a/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/OnnxFeatureConverter.java b/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/OnnxFeatureConverter.java
index ab143f77b6a..ec517768ea9 100644
--- a/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/OnnxFeatureConverter.java
+++ b/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/OnnxFeatureConverter.java
@@ -54,10 +54,10 @@ public class OnnxFeatureConverter extends ExpressionTransformer<RankProfileTrans
 
     private FeatureArguments asFeatureArguments(Arguments arguments) {
         if (arguments.isEmpty())
-            throw new IllegalArgumentException("An onnx node must take an argument pointing to " +
-                                               "the onnx model directory under [application]/models");
+            throw new IllegalArgumentException("An ONNX node must take an argument pointing to " +
+                                               "the ONNX model file under [application]/models");
         if (arguments.expressions().size() > 3)
-            throw new IllegalArgumentException("An onnx feature can have at most 2 arguments");
+            throw new IllegalArgumentException("An onnx feature can have at most 3 arguments");
 
         return new FeatureArguments(arguments);
     }