diff options
author | Lester Solbakken <lesters@users.noreply.github.com> | 2019-07-09 12:12:15 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2019-07-09 12:12:15 +0200 |
commit | 6e667b63879e23f9ec74817d6dc2ba3bf124bb93 (patch) | |
tree | 7d730892d2857085b009d97a599c88cb8e6f95cc /model-integration/src/test | |
parent | d9f23398047d5ff9fe9c068cc58fc2e310394c93 (diff) | |
parent | fde3e35a631fdfdbcf41a2466a65712b9a3f5ee2 (diff) |
Merge pull request #9996 from vespa-engine/havardpe/propagate-float-cell-type-and-avoid-using-it-too-much-right-now
propagate float and stop using it too much
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 07814687dc6..424e4d6c57c 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 @@ -43,14 +43,14 @@ public class OnnxMnistSoftmaxImportTestCase { // Check inputs assertEquals(1, model.inputs().size()); assertTrue(model.inputs().containsKey("Placeholder")); - assertEquals(TensorType.fromSpec("tensor<float>(d0[],d1[784])"), model.inputs().get("Placeholder")); + assertEquals(TensorType.fromSpec("tensor(d0[],d1[784])"), model.inputs().get("Placeholder")); // Check signature ImportedMlFunction output = model.defaultSignature().outputFunction("add", "add"); assertNotNull(output); assertEquals("join(reduce(join(rename(Placeholder, (d0, d1), (d0, d2)), constant(test_Variable), f(a,b)(a * b)), sum, d2), constant(test_Variable_1), f(a,b)(a + b))", output.expression()); - assertEquals(TensorType.fromSpec("tensor<float>(d0[],d1[784])"), + assertEquals(TensorType.fromSpec("tensor(d0[],d1[784])"), model.inputs().get(model.defaultSignature().inputs().get("Placeholder"))); assertEquals("{Placeholder=tensor(d0[],d1[784])}", output.argumentTypes().toString()); } |