use withTransformedExpressions for wiring

1 files changed, 1 insertions, 14 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 84d0997b12e..af072c5b59a 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
@@ -55,20 +55,7 @@ public class InputRecorder extends ExpressionTransformer<InputRecorderContext> {
                 }
                 return transformChildren(t, childContext);
             }
-            if (f instanceof DynamicTensor d) {
-                for (var tf : d.cellGeneratorFunctions()) {
-                    if (tf instanceof TensorFunctionNode.ExpressionTensorFunction expr) {
-                        transform(expr.wrappedExpression(), context);
-                    }
-                }
-            }
-            if (f instanceof Slice s) {
-                for (var tf : s.selectorFunctions()) {
-                    if (tf instanceof TensorFunctionNode.ExpressionTensorFunction expr) {
-                        transform(expr.wrappedExpression(), context);
-                    }
-                }
-            }
+            node = t.withTransformedExpressions(expr -> transform(expr, context));
         }
         if (node instanceof CompositeNode c)
             return transformChildren(c, context);