summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorLester Solbakken <lesters@users.noreply.github.com>2022-12-14 09:15:51 +0100
committerGitHub <noreply@github.com>2022-12-14 09:15:51 +0100
commit84e7048acbc7750ab5096a9cc09480e9d520977d (patch)
treeb795420ba1051d185eb9f4ab798c737088da4555
parent6a5e9dc02ce357ef639203b78fb8affe0e0a9685 (diff)
parenta021eeb282e5fa29de5809df07d51101eec367da (diff)
Merge pull request #25240 from vespa-engine/lesters/fix-model-eval-batch-dim
Fix testBatchDimensionEvaluation test by forcing long format
-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);