summaryrefslogtreecommitdiffstats
path: root/model-integration/src/test
diff options
context:
space:
mode:
authorLester Solbakken <lesters@oath.com>2019-12-18 10:47:55 +0100
committerLester Solbakken <lesters@oath.com>2019-12-18 10:47:55 +0100
commit6af40601ac82204b3732096dee2ba23fe237e6e8 (patch)
tree69bca5b6a9aca95312859924ba49bcf7571f7975 /model-integration/src/test
parent2041886aa197dff3a68abe5e821401f37adaa6f4 (diff)
Add option to use default signature for onnx models
Diffstat (limited to 'model-integration/src/test')
-rw-r--r--model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/OnnxOperationsTestCase.java2
1 files changed, 1 insertions, 1 deletions
diff --git a/model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/OnnxOperationsTestCase.java b/model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/OnnxOperationsTestCase.java
index 6954abe5157..9927f4df04f 100644
--- a/model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/OnnxOperationsTestCase.java
+++ b/model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/OnnxOperationsTestCase.java
@@ -391,7 +391,7 @@ public class OnnxOperationsTestCase {
private TensorFunction optimizeAndRename(String opName, IntermediateOperation op) {
IntermediateGraph graph = new IntermediateGraph(modelName);
graph.put(opName, op);
- graph.outputs(graph.defaultSignature()).put(opName, opName);
+ graph.outputs(IntermediateGraph.defaultSignature()).put(opName, opName);
graph.optimize();
return op.function().get();
}