1 files changed, 31 insertions, 5 deletions
diff --git a/config-model/src/test/integration/onnx-model/searchdefinitions/test.sd b/config-model/src/test/integration/onnx-model/searchdefinitions/test.sd
index 0f0fa694e6f..6e9ba356293 100644
--- a/config-model/src/test/integration/onnx-model/searchdefinitions/test.sd
+++ b/config-model/src/test/integration/onnx-model/searchdefinitions/test.sd
@@ -14,7 +14,7 @@ search test {
     }
 
     onnx-model my_model {
-        file:  files/ranking_model.onnx
+        file:  files/model.onnx
         input  first_input: attribute(document_field)
         input  "second/input:0": constant(my_constant)
         input  "third_input": my_function
@@ -22,19 +22,25 @@ search test {
     }
 
     onnx-model another_model {
-        file:  files/ranking_model.onnx
+        file:  files/model.onnx
         input  first_input: attribute(document_field)
         input  "second/input:0": constant(my_constant)
         input  "third_input": another_function
         output "path/to/output:2": out
     }
 
+    onnx-model dynamic_model {
+        file:  files/dynamic_model.onnx
+        input  input: my_function
+        output output: my_output
+    }
+
     rank-profile test_model_config {
         function my_function() {
             expression: tensor(d0[2])(1)
         }
         first-phase {
-            expression: onnxModel(my_model).out
+            expression: onnxModel(my_model).out{d0:1}
         }
     }
 
@@ -49,7 +55,7 @@ search test {
             expression: my_function()
         }
         first-phase {
-            expression: onnxModel("files/ranking_model.onnx", "path/to/output:1")
+            expression: onnxModel("files/model.onnx", "path/to/output:1"){d0:1}
         }
     }
 
@@ -62,9 +68,29 @@ search test {
         }
         summary-features {
             onnxModel(another_model).out
-            onnxModel("files/ranking_model.onnx", "path/to/output:2")
+            onnxModel("files/summary_model.onnx", "path/to/output:2")
+        }
+    }
+
+    rank-profile test_dynamic_model {
+        function my_function() {
+            expression: tensor(d0[1],d1[2])(d1)
+        }
+        first-phase {
+            expression: onnxModel(dynamic_model){d0:0,d1:1}
         }
+    }
 
+    rank-profile test_dynamic_model_2 {
+        function my_function_2() {
+            expression: tensor(d0[1],d1[3])(d1)
+        }
+        function my_function() {
+            expression: my_function_2()
+        }
+        first-phase {
+            expression: onnxModel(dynamic_model){d0:0,d1:2}
+        }
     }
 
 }