sort arguments

1 files changed, 9 insertions, 3 deletions

diff --git a/model-evaluation/src/main/java/ai/vespa/models/handler/ModelsEvaluationHandler.java b/model-evaluation/src/main/java/ai/vespa/models/handler/ModelsEvaluationHandler.java
index 78addf0328a..ef04b6641e5 100644
--- a/model-evaluation/src/main/java/ai/vespa/models/handler/ModelsEvaluationHandler.java
+++ b/model-evaluation/src/main/java/ai/vespa/models/handler/ModelsEvaluationHandler.java
@@ -20,7 +20,10 @@ import java.io.OutputStream;
 import java.net.URI;
 import java.nio.charset.Charset;
 import java.nio.charset.StandardCharsets;
+import java.util.ArrayList;
 import java.util.Arrays;
+import java.util.Collections;
+import java.util.List;
 import java.util.Map;
 import java.util.Optional;
 import java.util.concurrent.Executor;
@@ -134,10 +137,13 @@ public class ModelsEvaluationHandler extends ThreadedHttpRequestHandler {
         cursor.setString("info", baseUrl(request) + model.name() + "/" + compactedFunction);
         cursor.setString("eval", baseUrl(request) + model.name() + "/" + compactedFunction + "/" + EVALUATE);
         Cursor bindings = cursor.setArray("arguments");
-        for (Map.Entry<String, TensorType> argument : evaluator.function().argumentTypes().entrySet()) {
+        var argTypes = evaluator.function().argumentTypes();
+        List<String> argNames = new ArrayList<>(argTypes.keySet());
+        Collections.sort(argNames);
+        for (String name : argNames) {
             Cursor binding = bindings.addObject();
-            binding.setString("name", argument.getKey());
-            binding.setString("type", argument.getValue().toString());
+            binding.setString("name", name);
+            binding.setString("type", argTypes.get(name).toString());
         }
     }