aboutsummaryrefslogtreecommitdiffstats
path: root/model-evaluation/src/main/java/ai/vespa/models/handler/ModelsEvaluationHandler.java
diff options
context:
space:
mode:
Diffstat (limited to 'model-evaluation/src/main/java/ai/vespa/models/handler/ModelsEvaluationHandler.java')
-rw-r--r--model-evaluation/src/main/java/ai/vespa/models/handler/ModelsEvaluationHandler.java12
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());
}
}