summaryrefslogtreecommitdiffstats
path: root/model-evaluation
diff options
context:
space:
mode:
authorLester Solbakken <lesters@oath.com>2022-12-13 13:43:44 +0100
committerLester Solbakken <lesters@oath.com>2022-12-13 13:43:44 +0100
commita021eeb282e5fa29de5809df07d51101eec367da (patch)
tree746d2c34b6d696b830ad038cba1c2627157a32a4 /model-evaluation
parent4d5ab2bbce5b48332d18ab0f19a05cdb84c0efdb (diff)
Fix testBatchDimensionEvaluation test by forcing long format
Diffstat (limited to 'model-evaluation')
-rw-r--r--model-evaluation/src/test/java/ai/vespa/models/handler/OnnxEvaluationHandlerTest.java2
1 files changed, 1 insertions, 1 deletions
diff --git a/model-evaluation/src/test/java/ai/vespa/models/handler/OnnxEvaluationHandlerTest.java b/model-evaluation/src/test/java/ai/vespa/models/handler/OnnxEvaluationHandlerTest.java
index a9a36abe337..cc3f2863068 100644
--- a/model-evaluation/src/test/java/ai/vespa/models/handler/OnnxEvaluationHandlerTest.java
+++ b/model-evaluation/src/test/java/ai/vespa/models/handler/OnnxEvaluationHandlerTest.java
@@ -113,11 +113,11 @@ public class OnnxEvaluationHandlerTest {
handler.assertResponse(url, 200, expected);
}
- @Ignore
@Test
public void testBatchDimensionEvaluation() {
Map<String, String> properties = new HashMap<>();
properties.put("input", "tensor<float>(d0[],d1[3]):{{d0:0,d1:0}:0.1,{d0:0,d1:1}:0.2,{d0:0,d1:2}:0.3,{d0:1,d1:0}:0.4,{d0:1,d1:1}:0.5,{d0:1,d1:2}:0.6}");
+ properties.put("format.tensors", "long");
String url = "http://localhost/model-evaluation/v1/one_layer/eval"; // output not specified
Tensor expected = Tensor.from("tensor<float>(d0[2],d1[1]):[0.6393113,0.67574286]");
handler.assertResponse(url, properties, 200, expected);