diff options
| author | Harald Musum <musum@verizonmedia.com> | 2022-11-04 09:49:51 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-11-04 09:49:51 +0100 |
| commit | 3c30ddf019dac90aef337262fe427b85080c5062 (patch) | |
| tree | 7ca5f6c9c5758d264194c4e7cd4fb1d2a2906a0b /model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/TestableModel.java | |
| parent | 496699607666839c75877ce5686daceb9e1af4e2 (diff) | |
Revert "Since large constants are always tensors avoid creating a string just…"
Diffstat (limited to 'model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/TestableModel.java')
| -rw-r--r-- | model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/TestableModel.java | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/TestableModel.java b/model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/TestableModel.java index 2682ee40d76..5d13697df06 100644 --- a/model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/TestableModel.java +++ b/model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/TestableModel.java @@ -63,8 +63,8 @@ public class TestableModel { static Context contextFrom(ImportedModel result) { TestableModelContext context = new TestableModelContext(); - result.largeConstants().forEach((name, tensor) -> context.put("constant(" + name + ")", new TensorValue(tensor))); - result.smallConstants().forEach((name, tensor) -> context.put("constant(" + name + ")", new TensorValue(tensor))); + result.largeConstants().forEach((name, tensor) -> context.put("constant(" + name + ")", new TensorValue(Tensor.from(tensor)))); + result.smallConstants().forEach((name, tensor) -> context.put("constant(" + name + ")", new TensorValue(Tensor.from(tensor)))); return context; } |