diff options
author | Arne Juul <arnej@yahooinc.com> | 2022-12-15 11:39:10 +0000 |
---|---|---|
committer | Arne Juul <arnej@yahooinc.com> | 2022-12-15 11:46:40 +0000 |
commit | b5994fafd8a92746cd4543ba8bd33175a377e291 (patch) | |
tree | 30225d7081a48a1a45f9b8afec17c378be82b269 /model-evaluation | |
parent | a5ccdfb0c8180c0ec98ec258a02615ae58c71641 (diff) |
sort arguments
Diffstat (limited to 'model-evaluation')
-rw-r--r-- | model-evaluation/src/main/java/ai/vespa/models/handler/ModelsEvaluationHandler.java | 12 |
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()); } } |