diff options
author | Lester Solbakken <lesters@oath.com> | 2022-12-13 13:43:44 +0100 |
---|---|---|
committer | Lester Solbakken <lesters@oath.com> | 2022-12-13 13:43:44 +0100 |
commit | a021eeb282e5fa29de5809df07d51101eec367da (patch) | |
tree | 746d2c34b6d696b830ad038cba1c2627157a32a4 /model-evaluation | |
parent | 4d5ab2bbce5b48332d18ab0f19a05cdb84c0efdb (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.java | 2 |
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); |