From baa9c48be8732564e00730efe680df26a8f47f4c Mon Sep 17 00:00:00 2001
From: Arne Juul <arnej@yahooinc.com>
Date: Mon, 20 Mar 2023 12:19:34 +0000
Subject: use withTransformedExpressions for wiring

---
 .../yahoo/schema/expressiontransforms/InputRecorder.java  | 15 +--------------
 1 file changed, 1 insertion(+), 14 deletions(-)

(limited to 'config-model/src/main/java/com')

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);
-- 
cgit v1.2.3