Propagate type of transformer token helper functions

1 files changed, 9 insertions, 1 deletions

diff --git a/config-model/src/test/integration/onnx-model/schemas/test.sd b/config-model/src/test/integration/onnx-model/schemas/test.sd
index eac6f650fef..5b440e80bed 100644
--- a/config-model/src/test/integration/onnx-model/schemas/test.sd
+++ b/config-model/src/test/integration/onnx-model/schemas/test.sd
@@ -99,6 +99,15 @@ search test {
         }
     }
 
+    rank-profile test_dynamic_model_with_transformer_tokens {
+        function my_function() {
+            expression: tokenTypeIds(2, attribute(document_field))
+        }
+        first-phase {
+            expression: onnx(dynamic_model){d0:0,d1:1}
+        }
+    }
+
     rank-profile test_unbound_model {
         function my_function() {
             expression: tensor(d0[1],d1[2])(d1)
@@ -108,5 +117,4 @@ search test {
         }
     }
 
-
 }