diff options
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); |