Merge pull request #9944 from vespa-engine/bratseth/output-immediate-graph

Bratseth/output immediate graph

1 files changed, 8 insertions, 0 deletions

diff --git a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Shape.java b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Shape.java
index 79f3012c327..57175092b5c 100644
--- a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Shape.java
+++ b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Shape.java
@@ -37,6 +37,11 @@ public class Shape extends IntermediateOperation {
         return true;
     }
 
+    @Override
+    public Shape withInputs(List<IntermediateOperation> inputs) {
+        return new Shape(modelName(), name(), inputs);
+    }
+
     private void createConstantValue() {
         if (!allInputTypesPresent(1)) {
             return;
@@ -50,4 +55,7 @@ public class Shape extends IntermediateOperation {
         this.setConstantValue(new TensorValue(builder.build()));
     }
 
+    @Override
+    public String operationName() { return "Shape"; }
+
 }