diff options
author | Jon Bratseth <bratseth@verizonmedia.com> | 2019-05-08 12:27:09 +0200 |
---|---|---|
committer | Jon Bratseth <bratseth@verizonmedia.com> | 2019-08-13 13:22:31 +0200 |
commit | b32202458cce6a00686fab7bac777b6cb9ee34de (patch) | |
tree | d4fb80f544a13442f56e5d7d30ec09e26c6f8fe3 /model-integration/src/test | |
parent | e15d87688f4da812e93500598fa653164b47b9bd (diff) |
Merge
Diffstat (limited to 'model-integration/src/test')
-rw-r--r-- | model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/OnnxMnistSoftmaxImportTestCase.java | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/OnnxMnistSoftmaxImportTestCase.java b/model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/OnnxMnistSoftmaxImportTestCase.java index 424e4d6c57c..68df59bf93f 100644 --- a/model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/OnnxMnistSoftmaxImportTestCase.java +++ b/model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/OnnxMnistSoftmaxImportTestCase.java @@ -31,13 +31,13 @@ public class OnnxMnistSoftmaxImportTestCase { Tensor constant0 = Tensor.from(model.largeConstants().get("test_Variable")); assertNotNull(constant0); - assertEquals(new TensorType.Builder().indexed("d2", 784).indexed("d1", 10).build(), + assertEquals(new TensorType.Builder(TensorType.Value.DOUBLE).indexed("d2", 784).indexed("d1", 10).build(), constant0.type()); assertEquals(7840, constant0.size()); Tensor constant1 = Tensor.from(model.largeConstants().get("test_Variable_1")); assertNotNull(constant1); - assertEquals(new TensorType.Builder().indexed("d1", 10).build(), constant1.type()); + assertEquals(new TensorType.Builder(TensorType.Value.DOUBLE).indexed("d1", 10).build(), constant1.type()); assertEquals(10, constant1.size()); // Check inputs |